【YBTOJ进阶训练指导】选数游戏【二叉堆】【贪心】

最新推荐文章于 2023-03-02 00:04:16 发布

原创最新推荐文章于 2023-03-02 00:04:16 发布 · 3.7k 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#贪心算法 #题解 #c++ #YBTOJ #二叉堆

收录于

题解同时被 3 个专栏收录

478 篇文章

订阅专栏

数据结构

51 篇文章

订阅专栏

堆

10 篇文章

订阅专栏

本文介绍了如何解决YBTOJ中的选数游戏问题，采用贪心算法，按列依次选取数字，当空间不足时通过移除最小列进行优化。文章详细阐述了实现过程并提供了C++代码示例。

在这里插入图片描述

思路：

我们考虑贪心，一列一列加，如果不够空间了，就考虑去掉最小的那一列

$c o d e$

#include<iostream>
#include<cstdio>
#include<queue>

using namespace std;

long long ans, n, m, k, tmp, sum;
priority_queue<long long> q;

int main()
{
	scanf("%lld%lld%lld", &n, &m, &k);
	for(long long i=1; i<=m; i++)
	{
		long long x, maxx=0;
		scanf("%lld", &x);
		if(n==1)
		{
			if(i>k)
				continue;
			ans+=x;
			continue;
		}
		q.push(-x);
		tmp+=n, sum+=n*x;
		while(tmp>k&&!q.empty())
		{
			long long y=-q.top();
			q.pop();
			tmp-=n-1;
			sum-=(n-1)*y;
			maxx=max(maxx, y);
		}
		ans=max(ans, sum+min(n, k-tmp)*maxx);
		if(k<=i)
			break;
	}
	printf("%lld", ans); 
	return 0;
}