E. New Game Plus!
题意:
有n个经验值,怪物打的顺序可以任意。
题解:
本题只用贪心。。。。qwq
- 首先可以想到按经验值从大到小排序,先把正权的都取了。
- 什么时候停止呢?当总经验sum为负时。
- 这时候剩下的都是负经验值的怪,那么假如不使用清0。把剩下的负权分为k+1组。(k+1组是因为,有k次清0,且最后一个的不算)
- 现在考虑每个组应该应该什么时候使用技能?显然在一个组里,第1个被算了k次,第2个被算了k-1次。。。到第k+1个被算了0次。
- 那么对于后半部分就可以归于 ans += neg[i]*(i\k).(先从小到大排序)
反思:
- 是贪心,
- 在main里的自定应函数的复习。
AC
#include <iostream>
#include <vector>
#include <algorithm>
#define For(i,x,y) for(int i=(x); i<=(y); i++)
#define fori(i,x,y) for(int i=(x); i<(y); i++)
#define pb push_back
#define sz(a) (int)a.size()
using namespace std;
typedef long long ll;
const int maxn = 5e5+10;
ll a[maxn];
int main()
{
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
// int tt;cin>>tt;
// while(tt--){
vector<ll>neg;
int n, k; cin>>n>>k;
For(i,1,n)cin>>a[i];
// sort(a+1, a+1+n, greater<int>() );
sort(a+1, a+1+n, [&](int x, int y){return x>y;});
// For(i,1,n)cout<<a[i]<<' ';
// cout<<endl;
ll sum = 0, ans = 0;
int pos = n+1;
For(i,1,n){
ans += sum;
sum += a[i];
if(sum<0){
neg.pb(sum);
pos = i+1;
break;
}
}
// cout<<sum<<endl;
//cout<<ans<<endl;
For(i,pos,n)neg.pb(a[i]);
sort(neg.begin(),neg.end());
k++;
fori(i,0,sz(neg))ans += neg[i]*(i/k);
cout<<ans<<endl;
// }
return 0;
}