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<bits/stdc++.h>
#define LL long long
#define pa pair<LL,int>
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;
}