POJ3641 Pseudoprime numbers 快速幂

        这题比较简单,用快速幂进行加速幂运算。
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#include  
#include  
using namespace std;

long long qpowmod(long long x, int n, int mod)
{
	long long res = 1;
	while (n > 0)
	{
		if (n & 1)
		{
			res *= x;
			res %= mod;
		}

		x *= x;
		x %= mod;
		n >>= 1;
	}
	return res;
}

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

int main()
{
	int p, a;
#ifdef _DEBUG
	freopen("d:\\in.txt", "r", stdin);
#endif
	while (scanf("%d %d\n", &p, &a) != EOF)
	{
		if (p == 0 && a == 0)
		{
			break;
		}
		if (isprime(p))
		{
			printf("no\n");
			continue;
		}
		if (qpowmod(a, p, p) == a)
		{
			printf("yes\n");
		}
		else
			printf("no\n");
	}
	return 1;
}

你可能感兴趣的:(快速幂运算)