Educational Codeforces Round 108 (Rated for Div. 2) C Berland Regional

题意

有n个学校和n个学生，每个学生仅属于一个学校（可以有多个学生属于同一个学校），每个学生有一个技能值s，一个学校按技能值高低把该校的k个学生分为一组当一个组不满足k个人则忽略这些人，一个地区水平为各个学校所有组的学生技能值之和。求k（一个组的人数）从 1 - n 的对应的地区水平 n <= 2e5

输入

多组样例，先输入一个T表示样例总数，每个样例第一行输入一个n表示学校数和学生数，第二行输入n个数表示每个学生对应的学校，第三行输入n个数，表示每个学生的技能值

输出

对于每组样例输出n个数，第i个数表示k为i时的地区水平。

思路

当k个人为一组时，答案为各个学校的总技能值减去 每个学校最小的 学校人数%k个人的技能值之和，可以通过排序和前缀和解决，时间复杂度为O(nlogn)，之后的每一次操作为O(1)。如果对于每个k直接遍历所有的学校的时间复杂度为 O(n²)的，但是只有n个人意味着并不是所有的学校都有用，可以用一个unordered_set来维护学校信息，当k大于当前学校人数就可以无视也就是等价于删除，遍历每个学校只会遍历该学校的学生数次，所有学校遍历也就是等于总的学生个数次，最后处理的时间复杂度为O(n)

代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<unordered_map>
#include<unordered_set>
#include<set>
#include<map>
#include<queue>
#include<vector>

using namespace std;
#define x first
#define y second
#define int long long 
typedef pair<int ,int > PII;
typedef long long ll;
const int INF = 0x3f3f3f3f,mod = 1e9 + 7;
//const int N = ;
const int N = 2e5 + 10;
vector<int > nums[N];
unordered_set<int > s;
int a[N];
int sum[N];
signed main(){
	int T;
//	freopen("out.txt","w",stdout);
	cin >> T;
	while(T--){
//		int 
		int n;
		scanf("%lld",&n);
		for(int i = 1;i <= n;i++) scanf("%lld",&a[i]),s.insert(a[i]);
		for(int i = 1;i <= n;i++) {
			int x;
			scanf("%lld",&x);
			nums[a[i]].push_back(x);
		}
		for(auto x: s){
			sort(nums[x].begin(),nums[x].end());
			for(int i = 1;i < nums[x].size();i++){
				nums[x][i] += nums[x][i - 1];
			}
		}
		int v = 1;
		while(s.size())
		{
			vector<int>res;
			for(auto x:s){
				int t = nums[x].size();
				int u = t % v;
				int ss = nums[x].back();
				if(u >= 1) ss -= nums[x][u - 1];
				sum[v] += ss;
				if(v == t) res.push_back(x);  
			}
			for(auto x:res) s.erase(x);

			v ++;
		}
		for(int i = 1;i <= n;i++){
			printf("%lld ",sum[i]);
			nums[i].clear();
			sum[i] = 0;
		}
		puts("");
	}
	
	return 0;
}