【POJ】[3641]Pseudoprime numbers

#include<stdio.h>
bool prime(int m) {
for(int i=2; i*i<=m; i++)
if(m%i==0)
return false;
return true;
}
int pow(int a,int b) {
int MOD=b;
__int64 r=1,t=a%MOD;
while(b) {
if(b&1)
r=r*t%MOD;
t=t*t%MOD;
b>>=1;
}
return (int)r;
}
bool judge(int n,int m) {
if(prime(n))
return false;
if(pow(m,n)==m%n)
return true;
return false;
}
int main() {
int p,a;
while(scanf("%d %d",&p,&a),p||a) {
if(judge(p,a))
printf("yes\n");
else
printf("no\n");
}
return 0;
}