http://acm.hdu.edu.cn/showproblem.php?pid=1905
水题,套个米勒拉宾的模板就秒了
View Code
#include <stdio.h> __int64 qpow(int a,int b,int r)//快速幂 { __int64 ans=1,buff=a; while(b) { if(b&1)ans=(ans*buff)%r; buff=(buff*buff)%r; b>>=1; } return ans; } bool Miller_Rabbin(int n,int a)//米勒拉宾素数测试 { int r=0,s=n-1,j; if(!(n%a)) return false; while(!(s&1)){ s>>=1; r++; } __int64 k=qpow(a,s,n); if(k==1) return true; for(j=0;j<r;j++,k=k*k%n) if(k==n-1) return true; return false; } bool IsPrime(int n)//判断是否是素数 { int tab[]={2,3,5,7}; for(int i=0;i<4;i++) { if(n==tab[i]) return true; if(!Miller_Rabbin(n,tab[i])) return false; } return true; } int main() { int p,a ; while(scanf("%d%d",&p,&a),(p||a)) { if(IsPrime(p)) puts("no") ; else { if(qpow(a,p,p)==a) puts("yes") ; else puts("no") ; } } return 0 ; }