牛客练习赛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

输出

3

示例2

输入

3
1 3
2 3
100 1

输出

100

题解：

按si从大到小排序。用一个小根堆维护已选士兵的战力v。
设第i个必选，那么对团的人数起到限制作用的一定是si​。
因为si​是从大到小选的，所以已选士兵的s一定大于等于si。
那么选了第i个后，把战力小的踢出去，直到满足si的限制。
对所有结果取个max，就是答案。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
struct node{
    int x,y;
    bool operator < (const node A) const {
        return y>A.y;
    }
}a[maxn];
priority_queue<int,vector<int> ,greater<int> > q;
int main(){
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d%d",&a[i].x,&a[i].y);
    }
    sort(a,a+n);
    ll sum=0;
    ll ans=0;
    for(int i=0;i<n;i++){
        sum+=a[i].x;
        q.push(a[i].x);
        while(q.size()>a[i].y){
            sum-=q.top();
            q.pop();
        }
        ans=max(ans,sum);
    }
    printf("%lld\n",ans);
    return 0;
}