HDU 1905 Pseudoprime numbers

 

http://acm.hdu.edu.cn/showproblem.php?pid=1905

水题，套个米勒拉宾的模板就秒了

#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 ;
}