【快速幂】POJ3641 - Pseudoprime numbers

输入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=;
     for (int i=;i*i<=p;i++)
         if (p%i==)
         {
             f=;
             break;
         }
     return f;
 }

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

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