acm1905Pseudoprime numbers

做这道题时我还不知道有关于费马定理（或者我已经忘记），不知道怎么做，百度了一下，才知道这个定理。想一想和自己以前学的算法还是比较相像的，核心在于素数的判断以及暴力求解会溢出怎么解决的事
素数是直接判断，而a^p%p可以转化为代码中的解决方法
问题是：a^p%p=?a%p，并且p要求是素数
成立就yes,不成立为no

#include<stdio.h>
bool check(__int64 a){//检查是不是素数
    __int64 i;
    for(i=2;i*i<a;i++){
        if(a%i==0)
            return 0;
    }
    return 1;
}
__int64 fun(__int64 a,__int64 p,__int64 s){
    __int64 m;
    if(p==0)
        return 1;
    if(p%2==0){
        m=fun(a,p/2,s);
        return m*m%s;
    }
    else
        return  (a%s)*fun(a,p-1,s)%s;

}

int main(){
        __int64 p,a;
        while(scanf("%I64d %I64d",&p,&a)!=EOF){
            if(p==0&&a==0)
                break;
            if(check(p)){
                printf("no\n");
                continue;
            }   
            else{
                if(fun(a,p,p)==a)
                    printf("yes\n");
                else
                    printf("no\n");
            }   
        }
        return 0;
}