POJ 3641-Pseudoprime numbers

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

题意：p 不是素数时，a^p = a (mod p) ，则p 是符合的数

思路：快速幂运算+ 判段p 是否为素数。

#include <iostream>
#include<stdio.h>
using namespace std;
typedef long long ll;
bool no_prime(ll pp)
{
    int t=0;
    for(int i=2;i*i<pp;i++)
    {
        if(pp % i == 0) return true;
    }
    return false;
}

bool test(ll nn,ll r,ll pp)
{
    if(r% pp == nn) return true;
    return false;
}

ll mod_pow(ll nn,ll pp)
{
    ll res=1;
    ll mod=pp;
    while(pp > 0)
    {
        if(pp & 1) res=res * nn % mod;
        nn = nn * nn % mod;
        pp >>= 1;
    }
    //printf("%d\n",res);
    return res;
}
int main()
{
    ll p,n;
    while(~scanf("%lld%lld",&p,&n),n,p)
    {
        if(test(n,mod_pow(n,p),p) && no_prime(p)) printf("yes\n");
        else printf("no\n");
    }
    return 0;
}

CODE;