miller_rabin模板

miller_rabin素数测试法

#include <iostream>
#include <cstdlib>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <stdlib.h>
#include <iomanip>

#include <ctime>
#define ll long long
using namespace std;
ll mod_mul(ll a,ll b,ll n){
    ll res=;
    while(b){
        if(b&)res=(res+a)%n;
        a=(a+a)%n;
        b>>=;
    }
    return res;
}
ll mod_exp(ll a,ll b,ll n){
    ll res=;
    while(b){
        if(b&)res=mod_mul(res,a,n);
        a=mod_mul(a,a,n);
        b>>=;
    }
    return res;
}
bool miller_rabin(ll n){
    if(n==||n==||n==||n==||n==)return true;
    if(n==||!(n%)||!(n%)||!(n%)||!(n%)||!(n%))return false;
    ll x,pre,u;
    int i,j,k=;
    u=n-;
    while(!(u&)){
        k++;u>>=;
    }
    srand((ll)time());
    for(i=;i<;i++){//5次足够AC了
        x=rand()%(n-)+;
        if((x%n)==)continue;
        x=mod_exp(x,u,n);
        pre=x;
        for(j=;j<k;j++){
            x=mod_mul(x,x,n);
            if(x==&&pre!=&&pre!=n-)return false;
            pre=x;
        }
        if(x!=)return false;
    }
    return true;
}
int main(){
    ll n;
    while(cin>>n){
        if(miller_rabin(n))puts("is prime");
        else puts("is not prime");
    }
    return ;
}