Codeforces 980D Perfect Groups 计数

原文链接https://www.cnblogs.com/zhouzhendong/p/9074164.html

题目传送门 - Codeforces 980D

题意

　　$\rm Codeforces$ 真是个令人伤心的地方。

　　伤心的 $zzd$ 现在给你一个含有 $n$ 个数字元素的数列。

　　$zzd$ 问你对于 $1$ 到 $n$ 之间的每一个 $k$ 满足 $Q(序列)=k$ 的原序列的连续子序列个数。

　　其中，$Q()$定义如下：

　　把当前数列中的数分组，使得同组中任意两个数的乘积为完全平方数。其中最小分组数就是 $Q()$ 的值。

　　$n\leq 5000,-10^8\leq a_i\leq 10^8$

题解

　　伤心的 $zzd$ 再一次来到了令人伤心的 $\rm Codeforces $ ，再一次的看错题意，并再一次的没有考虑到坑点。

　　很容易发现对于一个数，它的平方因子对最后的分组没有影响。

　　我们把每一个数都除掉其最大平方因子，然后显然只有相同的数能分到同一组。

　　于是离散化一下 $O(n^2)$ 统计即可。

　　然后！！

　　有一个特殊的数字叫做 "0" ！

　　$0$可以随便分组。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=10005;
int n,a[N],cnt=0;
int pcnt=0,prime[N],f[N];
map <int,int> mp;
int tot,v[N],ans[N];
void get_prime(int n){
	for (int i=1;i<=n;i++)
		f[i]=1;
	f[1]=0;
	for (int i=2;i<=n;i++){
		if (!f[i])
			continue;
		prime[++pcnt]=i*i;
		for (int j=i*2;j<=n;j+=i)
			f[j]=0;
	}
}
int main(){
	get_prime(10000);
	scanf("%d",&n);
	mp.clear();
	for (int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		if (a[i]==0)
			continue;
		for (int j=1;j<=pcnt;j++)
			while (a[i]%prime[j]==0)
				a[i]/=prime[j];
		if (mp[a[i]]==0)
			mp[a[i]]=++cnt;
		a[i]=mp[a[i]];
	}
	for (int i=1;i<=n;i++){
		memset(v,0,sizeof v);
		tot=0;
		for (int j=i;j<=n;j++){
			if (a[j])
				tot+=v[a[j]]==0;
			v[a[j]]++;
			ans[max(tot,1)]++;
		}
	}
	for (int i=1;i<=n;i++)
		printf("%d ",ans[i]);
	return 0;
}