欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

牛客每日一题 3.25 tokitsukaze and Soldier (贪心+优先队列)

程序员文章站 2022-03-21 17:29:55
...

题目

牛客每日一题 3.25 tokitsukaze and Soldier (贪心+优先队列)

题解

枚举选择多少个物品,显然所有不低于此限制的物品均能选择,则使用一个优先队列储存当前选择的物品,先加入符合限制的,然后在删除容量的最小的部分即可。

代码

#include <iostream>
#include <queue>
#include <algorithm>
#define LL long long
using namespace std;
const int MAXN = 100020;
priority_queue<LL, vector<LL>, greater<LL> >q;//优先队列的使用,小根堆,删除最小的元素

struct Node {
	LL v;
	LL s;
}a[MAXN]; 

bool cmp(Node a, Node b) {
	return a.s > b.s;
}

int main() {
	int n;
	cin >> n;
	for(int i = 1; i <= n; i++) {
		cin >> a[i].v >> a[i].s;
	}
	sort(a+1, a+n+1, cmp);
	LL sum = 0;
	LL ans = 0;
	for(int i = 1; i <= n; i++) {
		sum += a[i].v;
		q.push(a[i].v);
		while(q.size() > a[i].s) {
			sum -= q.top();
			q.pop();
		}
		ans = max(sum, ans);
	}
	cout << ans << endl;
	return 0;
}