【Ybtoj 第2章 例题1】奶牛晒衣服【贪心】【大根堆】

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述


解题思路

注意到答案与最后一件烘干衣服所需要的时间有关。在不考虑烘干机的情况下,最后一件烘干的衣服定是湿度最大的衣服。所以我们希望湿度最大的衣服的湿度越小越好。

所以得到一个贪心策略:每次都选当前湿度最大的衣服使用烘干机。

模拟奶牛晒衣服的过程,设当前为第t分钟,若 a i − t ∗ A < = 0 a_i-t*A<=0 aitA<=0,则可得知第i件衣服被晒干。这样的时间复杂度是 O ( n 2 ) O(n^2) On2,所以我们可以用一个大根堆来维护衣服的湿度,把时间复杂度降到 O ( n l o g n ) O(n log n) Onlogn


代码

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
priority_queue<int> q;

int t,n,A,B,a[500010];

int main(){
    
    
	scanf("%d%d%d",&n,&A,&B);
	for(int i=1;i<=n;i++)
	{
    
    
		scanf("%d",&a[i]);
		q.push(a[i]);
	}
	while(q.top()-t*A>0)
	{
    
    
		t++;
		int x=q.top();
		q.pop();
		q.push(x-B);
	}	
	printf("%d\n",t);
}

猜你喜欢

转载自blog.csdn.net/kejin2019/article/details/111707422