2021-04-30

Educational Codeforces Round 108 (Rated for Div. 2)

C. Berland Regional

题目大意：首先给出一组数表示学校的编号，再给出一组数为对应学校选手的能力，然后给出整数k，一个学校每组成组成k人队伍，就派出这支队伍，能组成多支就派出多支队伍，当不足k人时就不派出队伍，然后各个学校所派出队伍的能力之和就表示该地区的能力，现要我们求出k为1~n时该地区该地区的最大能力各是多少？

思路：显然每个学校派出的队伍中的选手都时该学校中最强的，最先想到的可能就是开个二维数组将学校以及每个学校的人数存起，但数据很大很容易内存超限，而且很明显开个固定的二维数组浪费太大了，所以我们想到用vector来存储，这样可以优化一维极大的减少了浪费，然后我们将二维数组排序，然后对于不同的k先判断该学校的人数可以派出多少组，根据派出组数来挑选能力强的选手进行累加，将结果存入一个答案数组，但是会超时，由于选人是依次选取前面几个人而且是进行累加，这时应该联想到前缀和，用前缀和来优化。

代码：

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<queue>
#include<vector>
typedef long long ll;
using namespace std;

vector<ll>arr[200005];
ll n,ans[200005];
int db[200005];

bool cmp(ll a,ll b)
{
    return a>b;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(ans,0,sizeof(ans));
        int n,x;
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&db[i]);
            arr[i].clear();
        }
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&x);
            arr[db[i]].push_back(x);
        }
        for(int i=1; i<=n; i++)
        {
            sort(arr[i].begin(),arr[i].end(),cmp);
        }
        for(int i=1; i<=n; i++)
        {
            int len=arr[i].size();
            for(int j=1; j<len; j++)
            {
                arr[i][j]=arr[i][j-1]+arr[i][j];//前缀和
            }
        }
        for(int i=1; i<=n; i++)
        {
            int len=arr[i].size();
            for(int j=1; j<=len; j++)
            {
                ans[j]=ans[j]+arr[i][(len/j)*j-1];//保存答案
            }
        }
        for(int i=1; i<=n; i++)
            printf("%lld ",ans[i]);
        printf("\n");
    }
    return 0;
}