Miller Rabin 素数测试

Miller Rabin 素数测试

Miller Rabin 是一种快速判断素数的方法。有非常小的概率可能出错。

费马小定理和二次探测

Miller Rabin 素数测试主要就是用到了以上两个理论。

费马小定理: 如果$p$为质数，那么有$a^{p-1}\equiv 1modp$

二次探测: 对于质数$p$,如果$a^2\equiv 1modp$,那么$a\equiv 1$或$a\equiv p-1$

证明请自行百度。

判断$p$是否为质数

我们利用以上两个理论来进行素数判断

1. 如果$p$ 为偶数且不为$2$,我们可以直接判断出来

2. 将$p-1$分解成$2^k\times t$的形式，记录下$k,t$

3. 对于一个数$a$,我们首先求出$a^t$,然后进行二次探测，每次将$a$ 平方，二次探测$k$次

4. 如果以上全通过且满足费马小定理，那么我们判断$p$为质数

5. 只选一个$a$有很大概率出错，所以一般多选几个(竞赛用2,3,5,7,11,13,17,19,23即可)，且尽量选质数

注意:可能会爆long long，所以要采用快速乘。

期望时间复杂度:$O(n^{\frac{1}{4}})$

$\mathcal{C}ode$

/*******************************
Author:galaxy yr
LANG:C++
Created Time:2019年11月06日 星期三 19时24分57秒
*******************************/
#include<cstdio>
#include<algorithm>

using namespace std;

struct IO{
    template<typename T>
    IO & operator>>(T&res)
    {
        T q=1;char ch;
        while((ch=getchar())<'0' or ch>'9')if(ch=='-')q=-q;
        res=(ch^48);
        while((ch=getchar())>='0' and ch<='9') res=(res<<1)+(res<<3)+(ch^48);
        res*=q;
        return *this;
    }
}cin;

const int primes[]={2,3,5,7,11,13,17,19,23};
const int SIZE=9;

long long mul(long long a,long long b,long long mod)
{
    long long ans=0;
    while(b)
    {
        if(b&1)
            ans=(ans+a)%mod;
        a=(a+a)%mod;
        b>>=1;
    }
    return ans;
}

long long ksm(long long a,long long b,long long mod)
{
    long long res=1;
    while(b)
    {
        if(b&1)
            res=mul(res,a,mod);
        a=mul(a,a,mod);
        b>>=1;
    }
    return res;
}

bool Miller_Rabin(long long x)
{
    if(!(x&1))return false;
    long long tmp=x-1,s=0,k,nxt;
    while(!(tmp&1))s++,tmp>>=1;
    for(int i=0;i<SIZE;i++)
    {
        if(x==primes[i]) return true;
        k=ksm(primes[i],tmp,x);
        for(int j=1;j<=s;j++)
        {
            nxt=mul(k,k,x);
            if(nxt==1 && k!=1 && k!=x-1)
                return false;
            k=nxt;
        }
        if(k!=1) return false;
    }
    return true;
}