Codeforces Round #687 Problem - E.New Game Plus!（贪心）

Codeforces Round #687 Problem - E.New Game Plus!

题意

给了 $n$ 个数，有一个临时变量 $x$ ，初值为 $0$ ，每次可以任选一个数 $a_i$ ，将 $a_i$ 加到 $x$ 中，然后 $x$ 对答案做一次贡献。你有 $k$ 次清空 $x$ 的机会，问最终答案的最大值是多少。

解法

有 $k$ 次清空的操作，那么就可以看作这 $n$ 个数被分成了 $k+1$ 个组，假设一个组有 $b$ 个元素，那么第 $i$ 个元素对答案的贡献次数为 $b-i$ 。显然对同一个组来说大的数放前面更优，因为前面的数贡献大于后面的数。每次选择可贡献最大的组，增加一个元素，就可以增加最多的贡献。所以每次加一个数，都是加到$k+1$ 个组中和最大的一组，并统计贡献。

代码

#pragma region
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <vector>
using namespace std;
typedef long long ll;
#define tr t[root]
#define lson t[root << 1]
#define rson t[root << 1 | 1]
#define rep(i, a, n) for (int i = a; i <= n; ++i)
#define per(i, a, n) for (int i = n; i >= a; --i)
#pragma endregion
const int maxn = 5e5 + 5;
int n, k, a[maxn];
int main() {
    scanf("%d%d", &n, &k);
    rep(i, 1, n) scanf("%d", &a[i]);
    sort(a + 1, a + 1 + n, [](const int &A, const int &B) { return A > B; });
    priority_queue<ll> q;
    rep(i, 1, k + 1) q.push(0);
    ll ans = 0;
    rep(i, 1, n) {
        ll x = q.top();
        q.pop();
        ans += x;
        q.push(x + a[i]);
    }
    printf("%lld\n", ans);
}