POJ 3641 Pseudoprime numbers(快速幂+素数)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
bool isprime(int n)
{
bool flag=true;
for(int i=2;i*i<n;i++)
{
if(n%i==0)
{
flag=false;
break;
}
}
return flag;
}
LL POW(LL x,LL n,LL mod)
{
LL res=1;
while(n>0)
{
if(n&1) res=res*x%mod;
x=x*x%mod;
n>>=1;
}
return res;
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int p,a;
while(scanf("%d%d",&p,&a))
{
if(p==0&&a==0) break;
if(POW(a,p,p)==a&&!isprime(p))
{
printf("yes\n");
}
else
printf("no\n");
}
return 0;
}