原文:http://blog.csdn.net/sluqy671/article/details/41701655
首先了解一下伪素数的概念: 如果n是一个正整数,如果存在和n互素的正整数a满足a^(n-1)≡1(mod n),我们说n是基于a的伪素数。
如果一个数是伪素数,它几乎就是素数。另一方面,如果一个数不是伪素数,它一定不是一个素数。那么在一定的条件下,如果我们选取了若干个基都发现n是伪素数, 那么n是素数的概率趋近于1。
Miller-Rabbin测试
现在我们只需要多次寻找不超过n-1基并检验是否有a^(n-1)≡1(mod n), 如果一直有, 那么这个数就是一个素数, 否则可以立即判定这个数是个合数。
检验的时候使用快速幂进行优化, 可以使复杂度降至O(slogn), 这里s为选取基的次数。 可以证明, 出错的概率不大于1/(2^s), 因此s取到几十就差不多了。 再就是对于2要进行特判, 否则会对0求余, 那显然是不对的。
- #include <cstdio>
- #include <cmath>
- #include <cstdlib>
- #include <ctime>
-
- using namespace std;
-
- typedef long long LL;
- LL gcd(LL x, LL y)
- {
- if (!y) return x;
- return (y, x%y);
- }
- LL pow(LL a, LL x, LL mod)
- {
- LL ans = 1;
- while(x)
- {
- if (x & 1) (ans *= a) %= mod;
- (a *= a) %= mod;
- x >>= 1;
- }
- return ans;
- }
- bool MRT(LL x)
- {
- if (x == 2) return true;
- for (LL i = 1; i <= 30; ++i)
- {
- LL now = rand()%(x-2) + 2;
- if (pow(now, x-1, x) != 1) return false;
- }
- return true;
- }
- int main()
- {
- int n;
- LL x;
- scanf("%d", &n);
- while(n--)
- {
- scanf("%I64d", &x);
- if (MRT(x)) printf("YES\n");
- else printf("NO\n");
- }
- return 0;
- }