poj 3641 Fermat 与素数的判断

题意：给出p和a，判断p是否为非素数并且也满足a^p=a(mod p)，如果是，输出yes，否则输出 no。

简单的素数判断！，与fermat定理的应用。

#include<iostream>#include<cstdio>#include<cstring>typedef __int64 INT;using namespace std ;bool Isprime(int num){if((num>2&&num%2==0)||num==1) return 0;for(int i=3;i*i<=num;i+=2){if(num%i==0) return 0;}return 1;}INT Format_Mod(INT a,INT k,INT m){INT ans=1;while(k){if(k&1) ans=a*ans%m;a=(a*a)%m;k>>=1;}return ans;}int main(){INT a,p,res;    while(scanf("%I64d %I64d",&p,&a),(a||p)){if(!Isprime(p)){res=Format_Mod(a,p,p);if(res==a) puts("yes");else puts("no");}else puts("no");    } return 0 ;}