C. Playlist(STL+贪心)

C. Playlist

思路:

将数据按照歌曲的动听程度和长度的大小递减排序,首先保证歌曲的动听度最小,然后在不超过k个元素的情况下

使歌曲的长度之和最长,然后不断更新所得到的最大的愉悦度。

(这里可以用优先队列实现,题解上用的是set,但是有限队列注意是从小到大的)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = 300300;
typedef long long  LL;
priority_queue <LL,vector <LL>,greater <LL> > q;
struct Node{
	LL ti,bi;
}cur[maxn];
bool cmp(Node a,Node b){
	if(a.bi!=b.bi) return a.bi<b.bi;
	else return a.ti<b.ti;
}
LL MAX(LL x,LL y){
	return x>y?x:y;
}
int main(void)
{
	int n,k,i,j;
	LL ans = 0,sum = 0;
	scanf("%d%d",&n,&k);
	for(i=1;i<=n;i++){
		scanf("%lld%lld",&cur[i].ti,&cur[i].bi);
	}
	sort(cur+1,cur+1+n,cmp);
	while(!q.empty()) q.pop();
	for(i=n;i>=1;i--){
		q.push(cur[i].ti);
		sum += cur[i].ti;
		while(q.size()>k){
			//printf("top = %d\n",q.top());
			sum -= q.top();q.pop();
		}
		ans = MAX(ans,sum*cur[i].bi);
	}
	printf("%lld\n",ans);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41829060/article/details/89190595