【POJ】1811 Prime Test

http://poj.org/problem?id=1811

题意：求n最小素因子。（n<=2^54）

#include <cstdio>

#include <cstring>

#include <algorithm>

#include <cmath>

#include <iostream>

using namespace std;

typedef long long ll;

const ll lim=1e9;

inline void C(ll &a, ll c) { if(a>=c || a<=-c) a%=c; }

inline ll mul(ll a, ll b, ll c) { if(a<=lim && b<=lim) return a*b%c; ll x=0; for(; b; b>>=1, C(a+=a, c)) if(b&1) C(x+=a, c); return x; }

inline ll mpow(ll a, ll b, ll c) { ll x=1; for(; b; b>>=1, a=mul(a, a, c)) if(b&1) x=mul(a, x, c); return x; }

inline ll rand(ll a, ll b) {

	static const ll M=1e9+7, g=220703118;

	static ll now=1998;

	C(now*=g, M);

	return a+(now*now)%(b-a+1);

}

ll gcd(ll a, ll b) { return b?gcd(b, a%b):a; }

inline ll iabs(ll a) { return a<0?-a:a; }

inline ll PR(ll n, ll c) {

	ll x=rand(0, n-1), y=x, k=2, t;

	for(int i=2; ; ++i) {

		x=mul(x, x, n); x+=c; C(x, n);

		t=gcd(iabs(y-x), n);

		if(t!=1 && t!=n) return t;

		if(y==x) return n;

		if(i==k) y=x, k<<=1;

	}

}

bool check(ll n) {

	if(n==2 || n==3 || n==5 || n==7 || n==11 || n==13) return 1;

	if(n<2 || (n&1)==0 || n%3==0 || n%5==0 || n%7==0 || n%11==0 || n%13==0) return 0;

	ll d=n-1;

	int cnt=0;

	while((d&1)==0) d>>=1, ++cnt;

	for(int i=0; i<20; ++i) {

		ll a=mpow(rand(2, n-1), d, n);

		for(int i=0; i<cnt; ++i) { ll t=a; a=mul(a, a, n); if(a==1 && t!=1 && t!=n-1) return 0; }

		if(a!=1) return 0;

	}

	return 1;

}

ll f[100], ans;

int cnt;

void find(ll n) { //printf("%lld\n", n);

	if(check(n)) { 

		f[++cnt]=n; ans=min(ans, n);

		return;

	}

	ll p=n;

	while(p==n) p=PR(n, rand(1, n-1));

	find(p); find(n/p);

}



int main() {

	int T; scanf("%d", &T);

	while(T--) {

		ll n;

		scanf("%lld", &n);

		cnt=0; ans=n;

		find(n);

		if(cnt==1) puts("Prime");

		else printf("%lld\n", ans);

	}



	return 0;

}

　　

学习了下Pollard-Rho算法= =复杂度期望为$O(n^{1/4})$

具体不说看算导= =

大概就是用$x=x^2+c \pmod{n}$然后判$(y-x, n)$是否=1。其中$y$是第$2^k$个$x$。然后找到一个约数后递归这个约数和n/约数。（$c$和初始的$x$随机= =

然后如果碰到环退出就行了= =（一开始不知道为啥被卡了= =原来乘法爆掉了QAQ写个快速乘啊  ！！！