poj 3641 Pseudoprime numbers

#include <iostream>        //二分
using namespace std;
__int64 p;
bool prime(__int64 n)
{
    for(__int64 i=2;i*i<=n;++i)
        if(n%i==0)
            return false;
    return true;
}
__int64 query(__int64 a,__int64 n)
{
    if(n==1)
        return a;
    __int64 q=query(a,n/2),t=(q*q)%p;
    if(n%2==0)
        return t;
    else
        return (t*a)%p;
}
int main()
{
    __int64 a;
    while(cin>>p>>a&&p)
    {
        if(prime(p)||query(a,p)!=a)
            cout<<"no\n";
        else
            cout<<"yes\n";
    }
    return 0;
}