POJ-3641-Pseudoprime numbers

这个题要求求出若p不是素数并且p^a是否等于a，若等于则输出yes,否则输出no

这个题主要用到快速幂，其他就比较简单了

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
bool ispri(long long num)
{
    for(int i=2;i*i<num;i++)
	if(num%i==0)
	    return false;
    return true;
}
long long QuickMul(long long a,long long b)
{
    long long ans=1;
    long long m=b;
    while(m)
    {
	if(m&1)
	    ans=(ans*a)%b;
	a=(a*a)%b;
	m>>=1;
    }
    return ans;
}
int main()
{
    long long p,a;
    while(scanf("%lld%lld",&p,&a)&&(p+a))
    {
	if(ispri(p))
	    printf("no\n");
	else if(QuickMul(a,p)==a)
	    printf("yes\n");
	else
	    printf("no\n");
    }
    return 0;
}