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)。
输出描述:
输出一个正整数,表示团的最大战力。
示例1
输入
2
1 2
2 2
输出
3
示例2
输入
3
1 3
2 3
100 1
输出
100
我们来想一下这个题要怎么贪心,考虑到一个集合的士兵个数只与集合里面的最小的s[i]有关,那么我们就只需要枚举这个s[i],我们就能够得到所有的集合有可能的集合,那知道这些集合当中的s[i]之后,我只需要的就是在比s[i]大的士兵里面寻找s[i]-1个士兵其战斗力和最高即可,那我们对于s[i]从大到小排序,直接用优先队列维护!(难在贪心)

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<algorithm>
#define inf 100000
typedef long long ll;
using namespace std;
struct node{
    ll ft,s;
}data[inf+10];
bool cmp(struct node a,struct node b){
    return a.s>b.s;
}
struct opr{
    bool operator()(struct node a,struct node b){
        return a.ft>b.ft;
    }
};
int main (){
    ios::sync_with_stdio(0);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>data[i].ft>>data[i].s;
    }
    sort(data+1,data+1+n,cmp);
    priority_queue<node,vector<node>,opr> que;
    ll mmax=0,ans=0;
    for(int i=1;i<=n;i++){
        if(que.size()<data[i].s){
            que.push(data[i]);
            ans+=data[i].ft;
            mmax=max(mmax,ans);
        }else{
            while(que.size()>=data[i].s){
                ans-=que.top().ft;
                que.pop();
            }
            que.push(data[i]);
            ans+=data[i].ft;
            mmax=max(mmax,ans);
        }
    }
    cout<<mmax<<endl;
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章