后悔贪心法 牛客练习赛50（C题）

tokitsukaze and Soldier

题目描述

在一个游戏中，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)。

输出描述:

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

输入

2
1 2
2 2

输出

3

输入

3
1 3
2 3
100 1

输出

100

贪心+优先队列；
1.先把s[]排序，从大到小排序；
2.如果要取第 i 个人，那么人数肯定要小于等于s[i]；每次取 i 时，把 v[i] 放入一个小根堆中,当堆的大小大于s[i]，然后进行退队；然后记录每次的sum时，取最大值；

代码：

#include
#define LL long long
#define pa pair
using namespace std;
priority_queue< LL,vector<LL>,greater<LL> >qu;
struct Node{
	LL v,s;
}an[100100];
bool cmp(Node p,Node q){
	return p.s>q.s;
}
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%lld%lld",&an[i].v,&an[i].s);
	sort(an+1,an+n+1,cmp);
	LL sum=0;
	LL mmax=0;
	for(int i=1;i<=n;i++){
		sum+=an[i].v;
		qu.push(an[i].v);
		while(qu.size()>an[i].s){
			sum-=qu.top();
			qu.pop();
		}
		mmax=max(mmax,sum);
	}
	printf("%lld\n",mmax);
	return 0;
}