CF1468L - Prime Divisors Selection

CF1468L - Prime Divisors Selection

题目大意

对于一个序列\(A\),一个合法的质因子序列\(P\)满足\(\forall P_i|A_i,P_i\ is\ a\ prime\)

给定一个序列\(a_i,i\in[1,n]\),求选出\(k\)个数,使得对于选出的序列\(A\)

不存在一个\(P\)使得\(P\)中某个元素恰好出现一次

\(n\leq 1000,a_i\leq 10^{18}\)


分析

由题目的意思我们知道肯定要分解质因数

Pollard's_Rho!!!

\(10^{18}\)分解质因数可不是开玩笑的。。。

所以先考虑合法\(A\)的判定

判定\(A\)合法

假设可以存在一个元素恰好出现一次\(x\),那么在\(A\)所有元素质因子中至少要包含一个\(x\)

并且,不存在两个元素只包含\(x\)

也就是说,对于合法的\(A\)中出现的所有质因子\(x\),都必须存在两个元素只包含\(x\)

我们称只包含\(x\)作为质因子的元素为\(x\)-元素,为了构造合法的\(A\),我们必须对于一些\(x\)选出若干对\(x\)-元素

对于每个\(x\),我们只把前两个\(x-\)元素视为有效,假设有\(c\)对这样的元素

那么情况分几种

1.\(2|k,2c\ge k\),那么直接随意选完即合法

2.\(2c\ge k,2\not |k\),此时我们需要选出部分对,使得剩下的元素中存在一个数它的因子集已经被选

枚举剩下一个元素,判定合法即可

3.\(2c\leq k\),此时可以将\(c\)对全部选出,判断是否还存在\(k-2c\)个可以选择即可


质因数分解

亲身试验,我的Pollards_Rho它T飞了

容易发现,对于\(x-\)元素我们只需要找到它的\(a_i=x^k\)

对于其他元素我们只需要找到\(a_i\)对应的\(x\)的集合,或者判断无法被\(x-\)元素集合包含

由于\(n\leq 1000\),我们可以先得到\(x-\)元素集合,其他元素我们最后一个个判定

找到\(a_i=x^k\)问题简化了很多

如果你相信std::pow,可以直接来

只需要找到一个最小的\(k'\),使得\(a_i=x'^{k'}\),判定\(x'\)是否为质数,如果是则停止,否则继续分解\(x'\)

对于\(k'\leq 3\),甚至更大一些的情况,std::pow比较可信

而\(k'>3\)的情况(实际上\(k'=4\)被\(k'=2\)包含,所以是\(k'\ge 5\))

实际上\(x'\)已经很小了,直接枚举质数即可

素数判定依然需要\(\text{Miller_Rabin}\),但是至少不用Pollards_Rho了

CodeForces Submission

const int N=1e5+10;

int n,m;
int pri[N],pc,notpri[N];

ll qmul(ll x,ll y,ll P){
	ull z=(long double)x/P*y+0.5;
	ll res=(ull)x*y-z*P;  Mod2(res);
	return res;
}
ll qpow(ll x,ll k,ll P){
	ll res=1;
	for(;k;k>>=1,x=qmul(x,x,P)) if(k&1) res=qmul(res,x,P);
	return res;
}

int Miller_Rabin(ll n){
	if(n<N) return !notpri[n];
	if(~n&1) return 0;
	ll s=n-1,t=0;
	while(s%2==0) s/=2,t++;
	rep(k,1,7) {
		ll a=qpow(pri[rand()%pc+1],s,n),b;
		rep(i,1,t) {
			b=qmul(a,a,n);
			if(b==1 && a!=1 && a!=n-1) return 0;
			a=b;
		}
		if(a!=1) return 0;
	}
	return 1;
}

ll a[N],mk[N];
vector <ll> F[N]; // Factor Set of each element
vector <ll> IF; // Independent Factor Set
void unique(vector <ll> &a){ sort(a.begin(),a.end()),a.erase(unique(a.begin(),a.end()),a.end()); }

map <ll,vector<int> > M;
ll ans[N];
void Outp(){
	rep(i,1,m) ans[i]=a[ans[i]];
	sort(ans+1,ans+m+1);
	rep(i,1,m) printf("%lld ",ans[i]);
	exit(0);
}

ll Root2(ll n){
	ll x=round(sqrt(n));
	return x*x==n?x:-1;
}
ll Root3(ll n){
	ll x=round(pow(n,1./3));
	return x*x*x==n?x:-1;
}

ll KDivide(ll x){
	if(Miller_Rabin(x)) return x;
	ll y;
	if(~(y=Root2(x))) return KDivide(y);
	if(~(y=Root3(x))) return KDivide(y);
	ll U=pow(x,1./5)+4;
	vector <ll> fac;
	for(int i=1;pri[i]<=U;++i) if(x%pri[i]==0) {
		while(x%pri[i]==0) x/=pri[i];
		fac.pb(pri[i]);
	}
	if(fac.size()==1 && x==1) return fac[0];
	return -1;
}


int main(){
	rep(i,2,N-1) if(!notpri[i]) {
		pri[++pc]=i;
		for(int j=i+i;j<N;j+=i) notpri[j]=1;
	}
	n=rd(),m=rd();
	rep(i,1,n) {
		ll x=KDivide(a[i]=rd<ll>());
		if(~x) IF.pb(x);
	}
	unique(IF);
	rep(i,1,n) {
		ll x=a[i];
		for(ll y:IF) if(x%y==0) {
			while(x%y==0) x/=y;
			F[i].pb(y);
		}
		if(x>1) F[i].pb(-1),F[i].pb(-2); // invalid factor, emm... to avoid some situation we push two 
		if(F[i].size()==1 && M[F[i][0]].size()<2) M[F[i][0]].pb(i),mk[i]=1;
	}
	int c=0;
	for(auto i:M) if(i.second.size()>=2) c++;
	if(m%2==0 && c*2>=m) {
        // choose k/2 pairs!!
		int k=m;
		for(auto i:M) if(i.second.size()>=2) {
			if(!k) break;
			rep(j,0,1) ans[k--]=i.second[j];
		}
		Outp();
	}
	if(c*2>=m) {
        // find another
		rep(i,1,n) if(!mk[i] && (int)F[i].size()<=m/2) {
			int f=1;
			for(ll x:F[i]) if(M[x].size()<2) f=0;
			if(f) {
				int k=m;
				ans[k--]=i;
				for(ll x:F[i]) {
					rep(j,0,1) ans[k--]=M[x][j];
					M.erase(x);
				}
				for(auto i:M) if(i.second.size()>=2) {
					if(!k) break;
					rep(j,0,1) ans[k--]=i.second[j];
				}
				Outp();
			}
		}
	} else {
		int k=m;
		for(auto i:M) if(i.second.size()>=2) {
			if(!k) break;
			rep(j,0,1) ans[k--]=i.second[j];
		}
        // Count if we have left much enough...
		rep(i,1,n) if(!mk[i]) {
			if(!k) break;
			int f=1;
			for(ll x:F[i]) if(M[x].size()<2) f=0;
			if(f) ans[k--]=i;
		}
		if(!k) Outp();
	}
	puts("0");
}
上一篇:R语言分位数回归Quantile Regression分析租房价格


下一篇:CF1276F - Asterisk Substrings