Codeforces E. New Game Plus!(贪心)

传送门

题意：

有一个初始值$x=0$ 和总和$sum=0$，每次选择数组中的一个数$a[i]$ ,使得 $sum+=x$,$x+=a[i]$ 。

有$k$次重置初始值$x$的机会，即使 $x=0$ ,求最后能得到的最大的$sum$ 。每个数只能选择一次。

题解：

一开始肯定是先从大到小排序，然后一个一个取，直至$sum$ 变为负数。

当$sum$变为负数后，就需要使用那$k$ 次重置机会了。

其实$k$次重置机会就是将数组分为 $k+1$ 组，每组独立。那么也就是说我们要将当前的 $sum$ 和剩下的负数分为$k+1$ 组。

先来看看每组的贡献，假设长度为$len$ ,那么贡献就是 $b[1]\times (len-1)+b[2]\times (len-2)+...+b[len]\times 0$

这样就容易看出来了，我们肯定是要将最小的数乘以0的，然后再是乘以$1$ ,以此类推。

那么就是将剩下的数从小到大排序，每个数的贡献就是 $v[i]\times (i/k)$

代码：

#pragma GCC diagnostic error "-std=c++11"
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int> pii;
const int mod=1e9+7;
const int MAXN=5e5+5;
const int inf=0x3f3f3f3f;
ll a[MAXN];
vector<ll> v;
int main()
{
    int n, k;
    cin >> n >> k;
    for (int i = 1; i <= n;i++){
        cin >> a[i];
    }
    k++;
    ll sum = 0;
    ll ans = 0;
    sort(a + 1, a + 1 + n, greater<ll>());
    for (int i = 1; i <= n;i++){
        ans += sum;
        sum += a[i];
        if(sum<0){
            v.push_back(sum);
            for (int j = i + 1; j <= n;j++){
                v.push_back(a[j]);
            }
            break;
        }
    }
    sort(v.begin(), v.end());
    for (int i = 0; i < v.size();i++){
        ans += v[i] * (i / k);
    }
    cout << ans << endl;
}