Pseudoprime numbers(POJ-3641)(快速幂）

快速幂+素数判断

p必须不是素数。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
using namespace std;
typedef long long ll;
ll a,p;
ll mod_pow(ll x,ll n,ll mod) {
    ll res = 1;
    while(n>0) {
        if(n & 1) res = (res * x) % mod;
        x = (x * x) % mod;
        n >>= 1;
    }
    return res;
}
bool is_prime(int a) {
    for(int i=2;i*i<=a;i++)
        if(a%i==0) return false;
    return true;
}
int main() {
    while(~scanf("%lld%lld",&p,&a)) {
        if(a==0&&p==0) return 0;
        long long  v = mod_pow(a,p,p);
        if(v==a%p&&!is_prime(p)) printf("yes\n");
        else printf("no\n");
    }
    return 0;
}