hdu 1905 Pseudoprime numbers

#include<iostream>
#include<math.h>
using namespace std;


long long a,p;


bool PrimeNum(long long q) //判断素数
{
if(q==2)
return true;
if(!(q&1))
return false;
int j=sqrt(double(q));
for(long long i=3;i<=j;i+=2)
if(q%i==0)
return false;
return true;
}


long long func(long long a,int p) // 计算a^p mod p
{ 
long long k=1;
int mod=p;
while(p>=1)
{
if(p&1) //奇数除 补一位
k=k*a%mod;
a=a*a%mod;//根据p模2的次数，a指数增长
p>>=1;
}
return k;
}


int main()
{
while(cin>>p>>a,!(p==0&&a==0))
{
if(PrimeNum(p)) //先判断是否素数
{ 
cout<<"no"<<endl;
continue;
}
if(func(a,p)==a)
cout<<"yes"<<endl;
else cout<<"no"<<endl;
}
return 0;

}

注：用long long，否则会去高位，WR