HDU 1905 Pseudoprime numbers

题意：

给一个 p 和 一个 a，如果这个p 本身就是一个素数，就输出 no，如果不是素数，那么计算  ( a ^ p) % p 如果结果等于 a 那么输出 yes 否则输出 no。

快速幂取模就是在O(logn)内求出a^n mod b的值。算法的原理是(a*b) mod c=(a mod c)*(b mod c)mod c。

代码如下：

#include<iostream>
#include<cmath>

using namespace std;

int prime(long long n)
{
    int i;
    for (i=2; i<sqrt(n); i++)
    {
        if (n%i==0) return 0;
    }
    return 1;
}

long long mod(long long a, long long n, long long m)
{
    long long int t;
    if (n==0) return 1%m;
    if (n==1) return a%m;
    t=mod(a, n/2, m);
    t=t*t%m;
    if (n%2) t=t*a%m;
    return t;
}

int main()
{
    long long int a, p, i;
    while(cin >> p >> a, p||a)
    {
        if (prime(p))
        {
            cout << "no" << endl;
            continue;
        }
        long long k=mod(a, p, p);
        if (a == k)
            cout << "yes" << endl;
        else
            cout << "no" << endl;
    }
    return 0;
}