CF 980D Perfect Groups（数论）

CF 980D Perfect Groups（数论）

一个数组a的子序列划分仅当这样是合法的：每个划分中的任意两个数乘积是完全平方数。定义a的权值为a的最小子序列划分个数。现在给出一个数组b，问权值为i的b的子串个数。

这题意真不是人类智慧能轻易描述的。据说此题在比赛场上读题30min，做题5min，做完还WA。果然是坑题。

如果有两个数a和b，a和b的乘积是完全平方数，那么如果a有因子x^2，那么x2就可以去掉，使a变成a/x^2，结论依然成立。因此我们把所有数的质因子次数mod2，可以发现结论仍然不变。

因此，在一个划分内的数必须完全相同或者有数为零（涉及乘除法一定要考虑零啊啊啊）。n^2dp即可。

#include <cstdio>
#include <algorithm>
using namespace std;

const int maxn=5005;
int n, a[maxn], b[maxn], cnt[maxn];  //cnt[i]储存i有几个
int k, ans[maxn], id0;

int deal(int x){
	int flag;
	if (x<0) x=-x, flag=-1; else flag=1;
	for (int i=2; i*i<=x; ++i)
		while (x%(i*i)==0) x/=i*i;
	return x*flag;
}

int main(){
	scanf("%d", &n); int t; id0=-1;
	for (int i=1; i<=n; ++i){
		scanf("%d", &t), a[i]=b[i]=deal(t);
		if (!a[i]) id0=0;
	}
	sort(b+1, b+1+n);
	if (!id0) id0=lower_bound(b+1, b+1+n, 0)-b;
	for (int i=1; i<=n; ++i)
		a[i]=lower_bound(b+1, b+1+n, a[i])-b;
	for (int i=1; i<=n; ++i){
		for (int j=1; j<=n; ++j) cnt[j]=0; k=0;  //k:[i,j]不同元素的数目
		for (int j=i; j<=n; ++j){
			if (a[j]==id0){
				if (!k) ++ans[1]; else ++ans[k];
				continue;
			}
			if (!cnt[a[j]]) ++k;
			++cnt[a[j]]; ++ans[k];
		}
	}
	for (int i=1; i<=n; ++i) printf("%d ", ans[i]);
	return 0;
}