miller rabin 随机素数判定法模版

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>

using namespace std;
const int p[8]={3,5,7,11,13,17,19,23};//8个判定素数
int power_mod(int a,int b,int n)//快速幂
{
int i=0,ans=1;
while ((1<<i)<=b)
{
if (b&(1<<i)) ans=ans*a%n;
a=a*a%n;
i++;
}
return ans;
}
bool pan(int p,int n,int m,int k)
{
int pre=power_mod(p,m,n);//根据费马小定理
if (pre==1) return false;//如果a^n-1==1(mod n),那么它是一个素数，当然有伪素数，所以多判几遍
while (k--)
{
int next=pre*pre%n;//<span style="font-family: Arial; font-size: 16px; line-height: 21px;">而k次测试的基于费马小定理的一个推论：x^2 % n = 1当n为素数时x的解只有两个，即1和n-1。</span>
if (next==1&&pre!=1&&pre!=n-1) return true;
pre=next;
}
return (pre!=1);//根据上面所述，如果pre不等于1，那么这个数是合数
}
bool MR(int n)
{
if (n==2) return true;
if (n<2||!(n&1)) return false;

int k=0,m=n-1;
while (!(m&1)) m>>=1,k++;

for (int i=0;i<8;i++)
{
if (p[i]==n) return true;
if (pan(p[i],n,m,k)) return false;
}
return true;
}
int main()
{
int n;
scanf("%d",&n);
if (MR(n)) printf("Yes");else printf("No");
return 0;
}