poj3641

题目大意是这样的，输入p，a，两个数如果p是素数输出no，如果p不是素数，判断a^p%p==a是否成立，如果成立输出yes,否则输出no
很简单的事情，就是这么回事。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;

const long long MaxN = 1000000;
int prime[MaxN] = {0}, num_prime = 0;
int isNotPrime[MaxN] = {1 ,1};

void isprime()
{
    for(int i = 2 ; i < MaxN ; i++){
        if(!isNotPrime[i]) prime[num_prime++] = i;
        for(int j = 0 ; j < num_prime && i*prime[j] < MaxN ; j++){
            isNotPrime[i * prime[j]] = 1;
            if(!(i % prime[j])) break;
        }
    }
}
int judge(int p)
{
    int flag = 1;
    for(int i = 0 ; prime[i] < sqrt(p) ; i++)
        if(p % prime[i] == 0) flag = 0;
    if(flag) return 0;
    else return 1;
}
long long judgemi(long long p , long long a)
{
    long long res = 1 , n = p , m = a;
    while(n){
        if(n & 1) res = ( res * a ) % p;
        a = (a * a) % p;
        n >>= 1;
    }
    if(res % p == m) return 1;
    else return 0;
}
int main()
{
    isprime();
    //for(int i = 0 ; i < 50 ; i++) printf("%d-%d",i,prime[i]);
    int p ,a;
    while(~scanf("%d%d", &p, &a)){
        if(a == 0 && p == 0) break;
        if(judge(p)){
         //   printf("&&&\n");
            if(judgemi(p,a)) printf("yes\n");
            else printf("no\n");
        }
        else printf("no\n");
    }
    return 0;
}