[AGC003D]Anticube

题目

传送门 to AtCoder

思路

真是没想到啊。我直接排除了 “分解质因数” 的方法，因为 $\mathcal{O}(n\sqrt{s})$ 无法接受；谁知，不需要完全分解！

同样的，我感觉这是二分图。可是，我就没有想过，每个点只有一条邻边 吗？为什么我会一次次地视而不见？是什么蒙蔽了我的双眼？

首先，存在 $>\sqrt{s}$ 的质因数就不会连边了，因为找不出 $p^2>s$ 。更进一步，若存在两个不同的 $>\sqrt[3]{s}$ 的质因子，也不会连边。所以只把 $\sqrt[3]{s}$ 内的质因数分解出来，然后特判剩余部分是否为 $p$ 或 $p^2(\sqrt[3]{s}<p<\sqrt[2]{s})$ 。

复杂度 $\mathcal{O}(n\log n+n\frac{\sqrt[3]{s}}{\ln s})$，足以通过。

代码

卡模数：$998244353$ 和 $998244853$ 都被卡掉了……

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cctype>
#include <map>
using namespace std;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
typedef long long llong;
inline int readint(){
	int a = 0, c = getchar(), f = 1;
	for(; !isdigit(c); c=getchar())
		if(c == '-') f = -f;
	for(; isdigit(c); c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}

const int MAXN = 100005;
int primes[MAXN], primes_size;
bool isPrime[MAXN];
void sieve(int n){
	memset(isPrime+2,true,n-1);
	rep(i,2,n){
		if(isPrime[i]) primes[++ primes_size] = i;
		for(int j=1; j<=primes_size&&primes[j]<=n/i; ++j){
			isPrime[i*primes[j]] = false;
			if(i%primes[j] == 0) break;
		}
	}
}

int int_sqrt(llong n){
	llong x = 1; rep(i,1,20) x = (x+(n/x))>>1;
	while((x+1) <= n/(x+1)) ++ x;
	while(x > n/x) -- x; // manual adjustment
	return int(x); // Newton Iteration
}
const int MOD = 999999929, MAXV = 2200;
void split(llong n,int &me,int &he,int &big,int &cbig){
	for(int i=1,t=me=he=0; primes[i]<=MAXV; ++i){
		for(t=0; !(n%primes[i]); n/=primes[i]) ++ t;
		t %= 3, me = ((me<<1)%MOD+me+t)%MOD;
		he = ((he<<1)%MOD+he)%MOD; if(t) he += 3-t;
	}
	if(n == 1) return big = 0, void(cbig = 1); // self-loop
	big = 1, cbig = 1; // default: no edge to link
	if(n < MAXN && isPrime[n]) big = n, cbig = 1;
	else{
		int v = int_sqrt(n);
		if(isPrime[v] && llong(v)*v == n)
			big = v, cbig = 2;
	}
}

map<int,int> cnt[MAXN][2], bro[MAXN][2];
int main(){
	sieve(MAXN-5);
	int n = readint(), me, he, big, cbig, ans = 0;
	rep(i,1,n){
		llong x; scanf("%lld",&x);
		split(x,me,he,big,cbig); -- cbig;
		if(big == 1){ ++ ans; continue; } // big prime
		++ cnt[big][cbig][me], bro[big][cbig][me] = he;
	}
	if(cnt[0][0].count(0)) ++ ans, cnt[0][0].erase(0);
	for(auto it : cnt[0][0]){ // self-loop (no big prime)
		ans += max(it.second,cnt[0][0][bro[0][0][it.first]]);
		cnt[0][0].erase(bro[0][0][it.first]);
	}
	rep(i,1,MAXN-1) if(isPrime[i]) rep(j,0,1)
		for(auto it : cnt[i][j]){
			ans += max(it.second,cnt[i][j^1][bro[i][j][it.first]]);
			cnt[i][j^1].erase(bro[i][j][it.first]); // used
		}
	printf("%d\n",ans);
	return 0;
}