POJ 3641 Pseudoprime numbers（快速幂）

嗯...

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

AC代码：

#include<cstdio>
#include<iostream>

using namespace std;

inline bool is_prime(int x){
    if(x == 2) return 1;
    if(x % 2 == 0) return 0;
    for(int i = 3; i * i <= x; i += 2){
        if(!(x % i)) return 0;
    }
    return 1;
}

inline long long quick_mod(long long a, long long b, long long m){
    long long ans = 1;
    while(b){
        if(b & 1) ans = ans * a % m;
        a = a * a % m;
        b >>= 1;
    }
    return ans;
}

int main(){
    long long m, n;
    while(~scanf("%lld%lld", &m, &n) && m + n){
        if(is_prime(m)){
            printf("no
");
            continue;
        }
        long long ans;
        ans = quick_mod(n, m, m);
        if(ans == n) printf("yes
");
        else printf("no
");
    }
    return 0;
}