POJ3641-Pseudoprime numbers（快速幂取模）

题目大意

判断一个数是否是伪素数

题解

赤果果的快速幂取模。。。。

代码：

#include<iostream>
#include<cmath>
using namespace std;
#define LL long long
LL mul_mod(LL a,LL b,int n)
{
    return a*b%n;
}
LL pow_mod(LL a,LL p,LL n)
{
    if(p==0) return 1;
    LL ans=pow_mod(a,p/2,n);
    ans=ans*ans%n;
    if(p%2==1) ans=ans*a%n;
    return ans;
}
bool prime(LL n)
{
    if(n<2) return false;
    for(int i=2;i*i<n;i++)
    if(n%i==0) return false;
    return true;
}
int main()
{
    int a,p;
    while(cin>>p>>a&&p+a)
    {
        if(a%p==pow_mod(a,p,p))
        {
            if(!prime(p))
                cout<<"yes"<<endl;
            else
                cout<<"no"<<endl;
        }
        else
            cout<<"no"<<endl;
    }
    return 0;
}