CF EDU 108 C - Berland Regional

C - Berland Regional

算贡献

表面上若从 \(1\) 到 \(n\) 枚举 \(k\), 再枚举每个学校对第 \(k\) 个答案的贡献，复杂度为 \(O(n^2)\)

但因为一共只有 \(n\) 个学生，设每个学校有 \(cnt_i\) 个学生，那么每个学校最多只会对前 \(cnt_i\) 个答案有贡献，所以所有学校加一起对这 \(n\) 个答案只有 \(n\) 个贡献

所以可以把每个学校的学生从大到小排序，求出前缀和

然后枚举每个学校，求出这个学校对第 \(1\) 到 \(cnt_i\) 个答案的贡献，为 \(G[i] [len/k*k-1]\) (-1 是因为下标从 0 开始的)

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cmath>

using namespace std;
typedef long long ll;
const int N = 2e5 + 10;
vector<ll> G[N];
int n;
int id[N], s[N];
ll ans[N];
int main()
{
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	int T;
	cin >> T;
	while(T--)
	{
		cin >> n;
		for (int i = 1; i <= n; i++)
		{
			G[i].clear();
			ans[i] = 0;
		}
		for (int i = 1; i <= n; i++)
			cin >> id[i];
		for (int i = 1; i <= n; i++)
			cin >> s[i];
		for (int i = 1; i <= n; i++)
			G[id[i]].push_back(s[i]);
		for (int i = 1; i <= n; i++)
		{
			sort(G[i].begin(), G[i].end(), greater<ll>());
			for (int j = 1; j < G[i].size(); j++)
				G[i][j] += G[i][j-1];
		}	
		for (int i = 1; i <= n; i++)
		{
			int len = G[i].size();
			for (int k = 1; k <= len; k++)
				ans[k] += G[i][len/k*k-1];
		}
		for (int i = 1; i <= n; i++)
			cout << ans[i] << " ";
		cout << endl;
	}
	return 0;
}