Miller-Rabin素数测试

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long LL;
const LL prime[12]={2,3,5,7,11,13,17,19,23,29,31,37};//结论：对于Long Long范围的素数只需要这些
LL mul(LL a,LL b,LL n){   //********************蒙哥马利算法，解决a*b%n的问题，将b二进制拆开“快速加”，避免了a*b溢出
LL res=0;
while (b){
if (b&1) res=(res+a)%n;
a=(a+a)%n;
b>>=1;
}
return res;
}
LL pow(LL a,LL m,LL n){
LL res=1;
while (m){
if (m&1) res=mul(res,a,n);
a=mul(a,a,n);
m>>=1;
}
return res;
}

bool Miller_Rabin(LL p){
if (p==2) return 1;
if (p==1 || p%2==0 ) return 0;
LL k=0,e=p-1,s,i,j;
while (e%2==0) k++,e/=2;
for (i=0;i<12;i++){
if (prime[i]==p) return 1;
s=pow(prime[i],e,p);
for (j=1;j<=k;j++){
if (mul(s,s,p)==1 && !(s==1 || s==p-1)) return 0;//***************只有当X^2=1(MOD P)时才要二次剩余检验
s=mul(s,s,p);
}
if (s!=1) return 0;//fermat检验
}
return 1;
}
int main()
{
int t;
LL x;
scanf("%d",&t);
while (t--) {
scanf("%lld",&x);//****************************注意读入类型
Miller_Rabin(x) ? printf("Yes"):printf("No");
printf("\n");
}
return 0;
}

误判概率

经过独立的t轮Miller-Rabin算法将一个合数误判为素数的可能性不大于 (1/4)t，(正确实现的算法不会误判素数为合 数)，这个误判概率在基于Fermat定理的算法中是最好的。

　　这个误判概率是利用下面两个定理证明的：

　　定 理 1 ：设d=gcd(k,m) ，那么在有限群{g,g2,g3,?,gm=1}中(g是有限群的生成元，m是有限群的阶) 有d个元素x满足方程式xk=1。

　　定 理 2 ： 设p是一个奇素数，p-1=2sh(h是奇数)，那么在乘法群(Z/pZ)*中满足方程式 x2rt=-1mod p(t是奇数 )的元素的个数是：0，如果r ≥s；2rgcd(h,t)，如果r<s。 利用这两个定理，对算法输入n分3种情况讨论：

　　(1)n可以被某个素数的平方整除的时候;

　　(2)n是两个不同素数的乘积的时候;

　　(3)n是两个以上不同素数乘积的时候。

　　这样可以证明Miller-Rabin算法的误判概率上界。

优化实现

Miller-Rabin算法最为耗时的步骤在2.2模幂操作和2.3.2 循环。对算法的优化实现主要集中在对这两部分运算的优 化。对模幂操作的优化有两种途径：减少模幂算法中的模乘 操作和优化模乘操作。在求模幂的过程中不能先求幂最后一次求模，这样会产生一个十分巨大的中间结果，造成实际的 不可操作，所以在求模幂的算法中用模乘代替乘法，使得中 间结果的长度不超过模的长度。对模幂算法的优化，我们使 用改进的滑动窗口算法结合Montgomery模乘和模平方算法。

　　表1给出模幂算法的比较。

模幂算法	预先计算	模平方	模乘法
	模平方	模乘法	最坏情况	平均情况
平方乘算法 　　滑动窗口类算法 改进的滑动窗口算法	0 　　1 　　1	0 　　2k -3 　　2k-1-1	t 　　t-(k-1)≤次数≤t t-(k-1)≤次数≤t	t (t/k)-1 (t/k)-1	t/2 t/k(2k-1)/ 2k 　　k 　　≤t/k(2 -1)/

* 模幂算法比较，其中k是窗口大小，根据情况 选择以达到最优，t是指数的二进制位数。 优化的模幂算法描述：

　　输入： x，e=(e tet-1?e1e0)2，其中et=1，k≥1( 窗口大小)

　　输出： xe mod n

　　1、预计算

　　1.1、x1← MontMul(x， R2，n)，

　　x2←MontSqu(x 1， n)

　　1.2、对i 从1 到2k-1-1计算x2i+1←MontMul(x2i-1， x2，n)

　　2、A←R，i ←t

　　3、 当i≥ 0时作下面的操作：

　　3.1、如果ei=0，A←MontSqu(A ，n)，i← i-1

　　3.2、否则找到最长的位串eiei-1?es使得i-s+1≤k并且es=1,计算

　　3.2.1、A <－A2i-s+1 ， (利 用MontSqu函数计算)

　　3.2.2、A <－A*X(ee ...e )2 ，(利 用MontMul函数计算)

　　3.2.3、i ←s-1

　　4、A←MontMul(A ，1 ，n)

　　5、返回A

　　其中MontMul(x,y,n) 是Montgomery模乘函数，函数输出 结果为x*y*R-1 mod n，MontSqu(x,n) 是Montgomery模平方函 数，输出结果为x2R-1 mod n。模乘算法如果采用大整数乘法 和除法求模乘，因为涉及耗时的除法操作，所以要相对较 慢，结合大整数乘法和Barrett求模算法可以用2(n2+3n+1) 步 单精度乘法完成。使用Montgomery求模算法结合大整数乘法 算法，可以 在 2n(n+1) 步单精度乘法内完成算法。 Montgomery模平方的操作可以在3n(n+1)
/2步单精度乘法内 完成，而Barrett模平方需要(3n(n+3)/2+1) 步单精度乘法。结 合改进的滑动窗口算法和Montgomery类算法，可以得到目前 非特殊情况下的最优的模幂算法。

　　在Miller-Rabin算法的2.3.2循环中的模平方操作我们没有 使用Montgomery模平方算法，因为该算法给出的结果带有R-1这个参数，在2.3.2循环中处理掉这个参数将占整个循环运 行时间中的很大部分，尤其是在循环的控制参数s 相对较小的时候。我们在这里使用大整数平方算法结合Barrett求模算 法，2.3.2的循环最坏情况需要(s-1)(3n(n+3)/2+1)步单精度乘法。