【刷题】BZOJ 3667 Rabin-Miller算法

Input

第一行：CAS,代表数据组数（不大于350），以下CAS行，每行一个数字，保证在64位长整形范围内，并且没有负数。你需要对于每个数字：第一，检验是否是质数，是质数就输出Prime 
第二，如果不是质数，输出它最大的质因子是哪个。

Output

第一行CAS(CAS<=350，代表测试数据的组数) 
以下CAS行：每行一个数字，保证是在64位长整形范围内的正数。 
对于每组测试数据：输出Prime，代表它是质数，或者输出它最大的质因子，代表它是和数

Sample Input

6
2
13
134
8897
1234567654321
1000000000000

Sample Output

Prime
Prime
67
41
4649
5

HINT

数据范围：

保证cas<=350，保证所有数字均在64位长整形范围内。

Solution

裸Pollard Rho题

但它不简单，反而很恶心

不知道为什么数据那么强

几个注意的：

1）乘法要写快速乘，原理是a%b=a-a/b*b

2）Miller Rabin最好优化

3）有些版本的Pollard Rho是错的。。。被坑了好久（数学一本通）

这东西本身有概率错误，导致调都不知道调哪里，最后是照着zhou888的代码一点一点边改边交边调的

#include<bits/stdc++.h>
#define ll unsigned long long
const int Count=;
int base[]={,,,,,};
ll ans;
template<typename T> inline void read(T &x)
{
    T data=,w=;
    char ch=;
    while(ch!='-'&&(ch<''||ch>''))ch=getchar();
    if(ch=='-')w=-,ch=getchar();
    while(ch>=''&&ch<='')data=((T)data<<)+((T)data<<)+(ch^''),ch=getchar();
    x=data*w;
}
template<typename T> inline void write(T x,char c='\0')
{
    if(x<)putchar('-'),x=-x;
    if(x>)write(x/);
    putchar(x%+'');
    if(c!='\0')putchar(c);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
inline ll gcd(ll a,ll b)
{
    return b==?a:gcd(b,a%b);
}
inline ll qmul(ll a,ll b,ll n)
{
    return (a*b-(ll)(((long double)a*b+0.5)/n)*n+n)%n;
}
inline ll qexp(ll a,ll b,ll n)
{
    ll res=;
    while(b)
    {
        if(b&)res=qmul(res,a,n);
        a=qmul(a,a,n);
        b>>=;
    }
    return res;
}
inline bool Miller_Rabin(ll N)
{
    if(N==||(N>&&N%!=&&N%!=))return false;
    for(register int i=;i<=;++i)
        if(N==base[i])return true;
        else if(N%base[i]==)return false;
    ll p=N-,A,pre,k=;
    while(!(p&))p>>=,++k;
    for(register int i=;i<=Count;++i)
    {
        A=rand()%(N-)+;
        A=qexp(A,p,N);
        pre=A;
        for(register int j=;j<=k;++j)
        {
            A=qmul(A,A,N);
            if(A==&&pre!=&&pre!=N-)return false;
            pre=A;
        }
        if(A!=)return false;
    }
    return true;
}
inline ll abs(ll x,ll y)
{
    return y>x?y-x:x-y;
}
inline int Pollard_Rho(ll N,ll C)
{
    ll k=,x=rand()%N,y=x,d=;
    for(register ll i=;d==;++i)
    {
        x=(qmul(x,x,N)+C)%N;
        d=gcd(abs(x,y),N);
        if(i==k)k<<=1ll,y=x;
    }
    return d;
}
inline void solve(ll N)
{
    if(N==)return ;
    if(Miller_Rabin(N))
    {
        chkmax(ans,N);
        return ;
    }
    ll p,c=;
    while((p=Pollard_Rho(N,c))==N&&c<=N)c++;
    solve(p);solve(N/p);
}
int main()
{
    srand();
    int T;
    read(T);
    while(T--)
    {
        ll N;
        read(N);
        ans=;
        solve(N);
        if(ans==N)puts("Prime");
        else write(ans,'\n');
    }
    return ;
}

3667 Rabin-Miller算法