bzoj 3667: Rabin-Miller算法

模板题，但是感觉mul这个函数可能有问题。要不是数据范围是10^18，写快乘就行了。

     
     
     
     

      
      
      
      
 
       #include<cstdio> 
      
      
      
      
      
 
       #include<cstdlib> 
      
      
      
      
      
 
       #include<cstring> 
      
      
      
      
      
 
       #include<algorithm> 
      
      
      
      
      
 
         
      
      
      
      
      
 
       #define ll long long 
      
      
      
      
      
 
       #define inf 1e9 
      
      
      
      
      
 
       #define eps 1e-8 
      
      
      
      
      
 
       #define md 
      
      
      
      
      
 
       using namespace std; 
      
      
      
      
      
 
       ll mx; 
      
      
      
      
      
 
       ll mul(ll a,ll b,ll p) 
      
      
      
      
      
 
       { 
      
      
      
      
      
 
        ll tmp=(a*b-(ll)((long double)a/p*b+1e-8)*p); 
      
      
      
      
      
 
        return tmp<0?tmp+p:tmp; 
      
      
      
      
      
 
       } 
      
      
      
      
      
 
         
      
      
      
      
      
 
       ll kpow(ll a,ll b,ll p) 
      
      
      
      
      
 
       { 
      
      
      
      
      
 
        ll ans=1; 
      
      
      
      
      
 
        while (b) 
      
      
      
      
      
 
        { 
      
      
      
      
      
 
        if (b&1) ans=mul(ans,a,p); 
      
      
      
      
      
 
        a=mul(a,a,p); b>>=1; 
      
      
      
      
      
 
        } 
      
      
      
      
      
 
        return ans%p; 
      
      
      
      
      
 
       } 
      
      
      
      
      
 
         
      
      
      
      
      
 
       bool MR(int n,ll r,int s) 
      
      
      
      
      
 
       { 
      
      
      
      
      
 
        ll a=abs(rand())%(n-1)+1; 
      
      
      
      
      
 
        ll pre=kpow(a,r,n); 
      
      
      
      
      
 
        for (int i=1;i<=s;i++) 
      
      
      
      
      
 
        { 
      
      
      
      
      
 
        ll now=mul(pre,pre,n); 
      
      
      
      
      
 
        if (now==1&&pre!=n-1&&pre!=1) return 0; 
      
      
      
      
      
 
        pre=now; 
      
      
      
      
      
 
        } 
      
      
      
      
      
 
        if (pre!=1) return 0; 
      
      
      
      
      
 
        return 1; 
      
      
      
      
      
 
       } 
      
      
      
      
      
 
         
      
      
      
      
      
 
       bool is_prime(ll x) 
      
      
      
      
      
 
       { 
      
      
      
      
      
 
        if (x<2) return 0; 
      
      
      
      
      
 
        if (x==2) return 1; 
      
      
      
      
      
 
        if (!(x&1)) return 0; 
      
      
      
      
      
 
        int s=0; ll r=x-1; 
      
      
      
      
      
 
        while (!(r&1)) { r>>=1; s++;} 
      
      
      
      
      
 
        for (int i=0;i<10;i++) 
      
      
      
      
      
 
        if (!MR(x,r,s)) return 0; 
      
      
      
      
      
 
        return 1; 
      
      
      
      
      
 
       } 
      
      
      
      
      
 
         
      
      
      
      
      
 
       ll gcd(ll a,ll b) { return b==0?a:gcd(b,a%b);} 
      
      
      
      
      
 
         
      
      
      
      
      
 
       ll rho(ll n) 
      
      
      
      
      
 
       { 
      
      
      
      
      
 
        ll c=abs(rand())%(n-1)+1,x=abs(rand())%n,y=x; 
      
      
      
      
      
 
        int k=2; 
      
      
      
      
      
 
        for (int i=1;;i++) 
      
      
      
      
      
 
        { 
      
      
      
      
      
 
        x=(mul(x,x,n)+c)%n; 
      
      
      
      
      
 
        ll t=gcd(n,(y-x+n)%n); 
      
      
      
      
      
 
        if (t!=1&&t!=n) return t; 
      
      
      
      
      
 
        if (x==y) return n; 
      
      
      
      
      
 
        if (i==k) { k<<=1; y=x;} 
      
      
      
      
      
 
        } 
      
      
      
      
      
 
        return n; 
      
      
      
      
      
 
       } 
      
      
      
      
      
 
         
      
      
      
      
      
 
       void solve(ll x) 
      
      
      
      
      
 
       { 
      
      
      
      
      
 
        if (x==1) return; 
      
      
      
      
      
 
        if (is_prime(x)) { mx=max(mx,x); return; } 
      
      
      
      
      
 
        ll t=x; 
      
      
      
      
      
 
        while (t==x) t=rho(x); 
      
      
      
      
      
 
        solve(t); solve(x/t); 
      
      
      
      
      
 
       } 
      
      
      
      
      
 
         
      
      
      
      
      
 
       int main() 
      
      
      
      
      
 
       { 
      
      
      
      
      
 
        int tt; 
      
      
      
      
      
 
        scanf("%d",&tt); 
      
      
      
      
      
 
        while (tt--) 
      
      
      
      
      
 
        { 
      
      
      
      
      
 
        ll x; 
      
      
      
      
      
 
        scanf("%lld",&x); 
      
      
      
      
      
 
        mx=0; 
      
      
      
      
      
 
        solve(x); 
      
      
      
      
      
 
        if (mx!=x) printf("%lld\n",mx); 
      
      
      
      
      
 
        else printf("Prime\n"); 
      
      
      
      
      
 
        } 
      
      
      
      
      
 
        return 0; 
      
      
      
      
      
 
       }