CF1642B 题解

题目传送门

题目大意

有一个长度为 \(n\) 的数组 \(a\)，现在你需要把这个数组 \(n\) 个的元素分成 \(k\) 部分，每一段的对答案的贡献是这一部分不同的数的个数，最后的答案是所有的贡献之和。
现在需要求 \(k=1,2,\dots,n\) 的答案。
数据范围：\(n\le 3\times10^5,\sum n\le 3\times10^5,a_i\le10^9\)

题目解析

首先我们先求出整个序列有几个不同的数字，假设这个数字为 \(m\)。
当 \(k\le m\) 的时候，我们可以做到每个数字只出现在一个部分内，这样答案就是 \(m\)。
当 \(k>m\) 的时候，我们在 \(k=m\) 构造的基础上将 \(k-m\) 个元素单独放在一个部分内，答案就是 \(k\)。

核心代码：

int n,m,a[maxn];
void work(){
	n=read(); int i; for(i=1;i<=n;i++) a[i]=read(); sort(a+1,a+n+1); m=1;
	for(i=2;i<=n;i++) if(a[i]!=a[i-1]) m++;
	for(i=1;i<=n;i++)
		if(i<=m) print(m),pc(' ');
		else print(i),pc(' ');
	pc('\n'); return;
}