POJ 1811 Prime Test Miller-Rabin算法和Pollard rho算法

Description

Input

Output

Sample Input

2
5
10

Sample Output

Prime
2

Source

//400K	860MS
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#define Times 11
#define inf ((long long)1<<61)
#define C 201
long long jl[501],mini;//jl里面存的是大数的所有质因子
int ct;
long long gcd(long long a,long long b)//最大公约数
{
    return b==0?a:gcd(b,a%b);
}
long long random(long long n)//生成随机数
{
    return (long long)((double)rand()/RAND_MAX*n+0.5);
}
long long multi(long long a,long long b,long long m)//a*b%m
{
    long long ret=0;
    while(b>0)
    {
        if(b&1)ret=(ret+a)%m;
        b>>=1;
        a=(a<<1)%m;
    }
    return ret;
}
long long quick_mod(long long a,long long b,long long m)//a^b%m
{
    long long ans=1;
    a%=m;
    while(b)
    {
        if(b&1)
        {
            ans=multi(ans,a,m);
            b--;
        }
        b/=2;
        a=multi(a,a,m);
    }
    return ans;
}
bool Witness(long long a,long long n)
{
    long long m=n-1;
    int j=0;
    while(!(m&1))
    {
        j++;
        m>>=1;
    }
    long long x=quick_mod(a,m,n);
    if(x==1||x==n-1)return false;
    while(j--)
    {
        x=x*x%n;
        if(x==n-1)return false;
    }
    return true;
}
bool miller_rabin(long long n)//素数测试
{
    if(n<2)return false;
    if(n==2)return true;
    if(!(n&1))return false;
    for(int i=1; i<=Times; i++)
    {
        long long a=random(n-2)+1;
        if(Witness(a,n))return false;
    }
    return true;
}
long long pollard_rho(long long n,int c)//整数分解
{
    long long x,y,d,i=1,k=2;
    x=random(n-1)+1;
    y=x;
    while(1)
    {
        i++;
        x=(multi(x,x,n)+c)%n;
        d=gcd(y-x,n);
        if(1<d&&d<n)return d;
        if(y==x)return n;
        if(i==k)
        {
            y=x;
            k<<=1;
        }
    }
}
void find(long long n,int k)
{
    if(n==1)return;
    if(miller_rabin(n))
    {
        jl[++ct]=n;
        if(n<mini)mini=n;
        return;
    }
    long long p=n;
    while(p>=n)p=pollard_rho(p,k--);
    find(p,k);
    find(n/p,k);
}
int main()
{
    int cas;
    long long n;
    scanf("%d",&cas);
    while(cas--)
    {
        ct=0;
        mini=inf;
        scanf("%lld",&n);
        if(miller_rabin(n))printf("Prime\n");
        else{find(n,C);printf("%lld\n",mini);}
    }
    return 0;
}