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

一、算法分析

基本思路就是枚举，还有利用vector的一些小技巧。注意的点是，最后的枚举方式要有一个小优化，不能外层对k进行枚举，如果那样会超时，因为两层循环都是n。而是应该开一个res数组，以组为单位去维护res数组。

二、代码及注释

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#define LL long long 
using namespace std;
const int N=200050;
vector<LL> s[N];
int n;
int u[N];
LL res[N];                                            //res of every group
int main(){
    
    int T;
    cin>>T;
    while(T--){
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d",&u[i]);
            s[i].clear();
            res[i]=0;
        }
        for(int i=1;i<=n;i++){
            int v;
            scanf("%d",&v);
            s[u[i]].push_back(v);
        }
        for(int i=1;i<=n;i++){
            sort(s[i].begin(),s[i].end());
            reverse(s[i].begin(),s[i].end());
            for(int j=1;j<s[i].size();j++) s[i][j]=s[i][j-1]+s[i][j];
        }
        for(int i=1;i<=n;i++){                       //枚举每个组
            int len=s[i].size();
            for(int j=1;j<=len;j++){                 //这个组j个j个出的情况
                int mod=len%j;
                res[j]+=s[i][len-1-mod];
            }
        }
        for(int i=1;i<=n;i++) printf("%lld ",res[i]);
        printf("\n");
    }
    
    
    
    
    return 0;
    
}