poj 3641 Fermat 与素数的判断

#include<iostream>
#include<cstdio>
#include<cstring>
typedef __int64 INT;
using namespace std ;

bool Isprime(int num)
{
if((num>2&&num%2==0)||num==1) return 0;
for(int i=3;i*i<=num;i+=2){
if(num%i==0) return 0;
}
return 1;
}

INT Format_Mod(INT a,INT k,INT m)
{
INT ans=1;
while(k){
if(k&1) ans=a*ans%m;
a=(a*a)%m;
k>>=1;
}
return ans;
}

int main()
{
INT a,p,res;
while(scanf("%I64d %I64d",&p,&a),(a||p)){
if(!Isprime(p)){
res=Format_Mod(a,p,p);
if(res==a) puts("yes");
else puts("no");
}
else puts("no");
}
return 0 ;
}