牛客-tokitsukaze and Soldier(貪心+stl優先隊列)

原題鏈接:更好的閱讀體驗
每日一題題單鏈接:傳送門
題目描述
在一個遊戲中,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]的話就要去掉,貪心一下肯定是去掉v[i]值小的。
如果每一次都sort的話複雜度可能不大樂觀,所以就可以用優先隊列~
隊列的詳解用法:傳送門
隊列與優先隊列的總結:傳送門
代碼:

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

友好的每日一題開端~

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