poj-3641(Pseudoprime numbers)

学习快速幂之后做的第一道题，这里记录一下。

题目大意：给出两个数 a 和 p，如果 p 是一个素数，判断是否满足公式 a^p %p=a，如果满足，则输出yes，否则，输出no。附链接：http://poj.org/problem?id=3641。

大体思路：题目简单，判断素数，然后判断是否满足公式。

以下是ac代码：

#include<iostream>
using namespace std;
long long power(long long a,long long b,long long mod){
    long long ans=1;
    long long temp=a;
    while(b!=0){
        if(b%2)
            ans=ans*temp%mod;
        temp=temp*temp%mod;
        b/=2;
    }
    return ans;
}
bool is_prime(long long a){
    if(a<=1)
        return false;
    for(long long i=2;i*i<=a;i++)
        if(a%i==0)
            return false;
    return true;
}
int main(){
    long long p,a;
    while(cin>>p>>a){
        if(p==0)
            break;
        if(is_prime(p)){
            cout<<"no"<<endl;
            continue;
        }
        if(power(a,p,p)==a)
            cout<<"yes"<<endl;
        else
            cout<<"no"<<endl;
    }
    return 0;
}