E - New Game Plus! 堆贪心 思维

Posted 2021-11-05 goto_1600

提交 1000祭
题意：

思路：
一开始想的就是按照值的大到小排序，然后不知道负数的时候怎么处理了，看了题解原来可以用堆，由于可以清k次零，也是就是把序列分割成k+1组，我们可以发现在每一组里面值是非递增的，否则可以交换两个元素，然后可以用堆维护每个组当前的值，每次取最大，然后把最大的值赋给当前最大的值。复杂度O(NlogN)
代码：

// Problem: CF1415E New Game Plus!
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/CF1415E
// Memory Limit: 250 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
typedef long long ll; 
using namespace std;
//#define int long long
const int N=500010;
int c[N];
ll sum[N];
ll sum2[N];
signed main()
{
	ios;
	int n,k;
	cin >> n >> k;
	for(int i=1;i<=n;i++)	cin>>c[i];
	
	sort(c+1,c+1+n,greater<int>());
	ll res=0;
	priority_queue<ll>q;
	for(int i=1;i<=k+1;i++)
		q.push(0);
	for(int i=1;i<= n ; i++)
	{
		auto t=q.top();
		q.pop();
		res+=t;
		t+=c[i];
		q.push(t);
	}
	cout<<res<<endl;
	return 0;
}