牛客每日一题 3.25 tokitsukaze and Soldier (贪心+优先队列)
程序员文章站
2022-03-21 17:29:55
...
题目
题解
枚举选择多少个物品,显然所有不低于此限制的物品均能选择,则使用一个优先队列储存当前选择的物品,先加入符合限制的,然后在删除容量的最小的部分即可。
代码
#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;
}