【快速幂】POJ3641

 输入a和p。如果p不是素数，则若满足a^p = a (mod p)输出yes，不满足或者p为素数输出no。最简单的快速幂，啥也不说了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
ll p,a; 

int whether(int p)
{
    int f=1;
    for (int i=2;i*i<=p;i++)
        if (p%i==0)
        {
            f=0;
            break;
        }
    return f;
}

int submain()
{
    ll res=1,n=p,x=a;
    while (n>0)
    {
        if (n&1) res=res * x % p;
        /*如果n最后一位是一，那么乘上x*/
        x=x*x % p;
        n>>=1;
        /*右移以为，即除以二*/
    }
    return (res==a);
}

int main()
{
    while (scanf("%lld%lld",&p,&a))
    {
        if (p==a && a==0) break;
        if (!whether(p))
        {
            if (submain()) cout<<"yes"<<endl;
                else cout<<"no"<<endl;
        }
        else
            cout<<"no"<<endl;
    } 
    return 0;
}