牛客练习赛50 C tokitsukaze and Soldier

链接：

https://ac.nowcoder.com/acm/contest/1080/C

来源：牛客网

题目描述

在一个游戏中，tokitsukaze需要在n个士兵中选出一些士兵组成一个团去打副本。
第i个士兵的战力为v[i]，团的战力是团内所有士兵的战力之和。
但是这些士兵有特殊的要求：如果选了第i个士兵，这个士兵希望团的人数不超过s[i]。(如果不选第i个士兵，就没有这个限制。)
tokitsukaze想知道，团的战力最大为多少。

输入描述:

第一行包含一个正整数n(1≤n≤10^5)。
接下来n行，每行包括2个正整数v,s(1≤v≤10^9,1≤s≤n)。

输出描述:

输出一个正整数，表示团的最大战力。

示例1

输入

复制

2
1 2
2 2

输出

复制

示例2

输入

复制

输出

复制

题解：

贪心，先按s从大到小排序，如果要选第i个士兵，如果已选的人数已经超过了s[i]，就把已选的人里面战力最小的去掉（用优先队列实现），遍历一遍找maxn（代码易懂

#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;

struct node
{
    int v, s;
}a[100010];
priority_queue<int, vector<int>, greater<int>>q;

bool cmp(node x, node y) {
    return x.s > y.s;
}
int main() {
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i++) scanf("%d %d", &a[i].v, &a[i].s);
    sort(a, a + n, cmp);
    long long sum = 0, maxn = 0;
    for(int i = 0; i < n; i++) {
        sum += a[i].v;
        while(q.size() >= a[i].s) {
            int b = q.top();
            q.pop();
            sum -= b;
        }
        q.push(a[i].v);
        if(sum > maxn) maxn = sum;
    }
    printf("%lld
", maxn);
    return 0;
}