POJ 3641 Pseudoprime numbers (快速幂取模)

http://poj.org/problem?id=3641

注意计算前先 a%=p 一下。

/*16ms,384KB*/

#include<cstdio>

long long qmod(long long a, long long n, long long b)
{
	long long ans = 1;
	while (n)
	{
		if (n & 1)
		{
			ans = (ans * a) % b;
		}
		a = (a * a) % b;
		n >>= 1;
	}
	return ans;
}

bool isprime(int p)
{
	for (int i = 2; i * i <= p; i++)
		if (p % i == 0)
			return false;
	return true;
}

int main()
{
	long long a, p;
	while (scanf("%lld%lld", &p, &a), p)
	{
		a %= p;
		if (!isprime((int)p) && (qmod(a, p, p) == a)) puts("yes");
		else puts("no");
	}
	return 0;
}