前缀和 + 暴力 + 排序 - Berland Regional - CF 1519C

前缀和 + 暴力 + 排序 - Berland Regional - CF 1519C

分析：

$说到底，本题是一个要求快速计算区间和的问题。$

$根据题意，每次需要对每个集合前m\times j大的数进行求和，$

$其中j\in [1,n]，是分组的长度；m是满足m\times k\le le{n}_i的最大的正整数；len是第i个集合的大小。$

$所以，我们对每个集合先从大到小进行排序。$

$下面的关键问题在于，如何快速计算m\times k的值：$

$也就是说，我们每次要去除末尾的，模k余下来的几个数。$

$$也就是前le{n}_i-le{n}_i\%j=\lfloor \frac{le{n}_i}{j}\rfloor \times j个数。$$

$由于我们使用vector来存储，下标从0开始，所以最后的下标还要相应-1。$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<unordered_map>

#define x first
#define y second

using namespace std;

typedef long long ll;
typedef pair<int, int> P;
const int N = 200010;

int t, n, u[N], x;
vector<ll> g[N];
ll ans[N];

void init()
{
    for(int i=1;i<=n;i++) 
    {
        ans[i] = 0;
        g[i].clear();
    }
}

int main()
{
    scanf("%d", &t);
    while(t --)
    {
        scanf("%d", &n);
              
        init();

        for(int i=1;i<=n;i++) scanf("%d", &u[i]);
        for(int i=1;i<=n;i++)
        {
           scanf("%d", &x);
           g[u[i]].push_back(x);
        }
        
        //降序
        for(int i=1;i<=n;i++) 
            sort(g[i].begin(), g[i].end(), greater<int>());
        
        //前缀和
        for(int i=1;i<=n;i++)
        {
            int m = g[i].size();
            for(int j=1;j<m;j++)
                g[i][j] += g[i][j-1];
        }

        //计算答案
        for(int i=1;i<=n;i++)
        {
            int len = g[i].size();
            for(int j=1;j<=len;j++)
                ans[j] += g[i][len/j*j-1];
        }
       
        for(int i=1;i<=n;i++) printf("%lld ",ans[i]);
        puts("");
    }
   
    return 0; 
}