判断2^p-1是不是梅森素数（Miller-Rabin素数测试）

 

判断2^p-1是不是梅森素数（Miller-Rabin素数测试）

分类： 数论 2012-08-27 21:16  866人阅读  评论(0)  收藏  举报

题目：梅森素数 

[cpp]  view plain copy

1. #include <iostream>  
2. #include <stdio.h>  
3. #include <stdlib.h>  
4.   
5. using namespace std;  
6. typedef long long LL;  
7.   
8. const int Times=10;  
9.   
10. LL multi(LL a,LL b,LL m)  
11. {  
12.     LL ans=0;  
13.     while(b)  
14.     {  
15.         if(b&1)  
16.         {  
17.             ans=(ans+a)%m;  
18.             b--;  
19.         }  
20.         b>>=1;  
21.         a=(a+a)%m;  
22.     }  
23.     return ans;  
24. }  
25.   
26. LL quick_mod(LL a,LL b,LL m)  
27. {  
28.     LL ans=1;  
29.     a%=m;  
30.     while(b)  
31.     {  
32.         if(b&1)  
33.         {  
34.             ans=multi(ans,a,m);  
35.             b--;  
36.         }  
37.         b>>=1;  
38.         a=multi(a,a,m);  
39.     }  
40.     return ans;  
41. }  
42.   
43. bool Miller_Rabin(LL n)  
44. {  
45.     if(n==2) return true;  
46.     if(n<2||!(n&1)) return false;  
47.     LL a,m=n-1,x,y;  
48.     int k=0;  
49.     while((m&1)==0)  
50.     {  
51.         k++;  
52.         m>>=1;  
53.     }  
54.     for(int i=0;i<Times;i++)  
55.     {  
56.         a=rand()%(n-1)+1;  
57.         x=quick_mod(a,m,n);  
58.         for(int j=0;j<k;j++)  
59.         {  
60.             y=multi(x,x,n);  
61.             if(y==1&&x!=1&&x!=n-1) return false;  
62.             x=y;  
63.         }  
64.         if(y!=1) return false;  
65.     }  
66.     return true;  
67. }  
68.   
69. int main()  
70. {  
71.     LL n,t;  
72.     cin>>t;  
73.     while(t--)  
74.     {  
75.         cin>>n;  
76.         n=((LL)1<<n)-1;  
77.         if(Miller_Rabin(n)) puts("yes");  
78.         else puts("no");  
79.     }  
80.     return 0;  
81. }