HDU 1905 Pseudoprime numbers

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=1905

题意：给一个 p 和 一个 a，如果这个p 本身就是一个素数，就输出 no，如果不是素数，那么计算  ( a ^ p) % p 如果结果等于 a 那么输出 yes 否则输出 no

也就是伪素数。

#include<stdio.h>
#include<math.h>
__int64 jud(__int64 n)
{
    __int64 i;
    for(i=2;i*i<n;i++)
        if(n%i==0)
        return 0;
   return 1;
}
__int64 sum(__int64 a,__int64 p,__int64 m)
{
    __int64 s;
    if(p==0)
        return 1;
    if(p%2==0)
    { 
        s=sum(a,p/2,m);
      return (s*s)%m; 
    }
    else
        return 
        (a%m)*sum(a,p-1,m)%m;

}
int main()
{
    __int64 p,a,i;
    while(~scanf("%I64d %I64d",&p,&a))
    {
        if(p==0&&a==0)
            break;
        if(jud(p))
        {
            printf("no\n");
            continue;
        }
        if(sum(a,p,p)==a)
            printf("yes\n");
            else
            printf("no\n");
    }
    return 0;
}