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

题目链接：http://poj.org/problem?id=3641

题意：给出a,p，如果p是素数且 a^p = a (mod p)即a的p次方对p取模余数为a，输出"yes",否则输出"no".

题解：快速幂+判断素数。

代码：

#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;
}