标签：cnt Berland int CF cin 学校 108 枚举 include

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;
}

标签：cnt,Berland,int,CF,cin,学校,108,枚举,include
来源： https://www.cnblogs.com/hzy717zsy/p/16282902.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。