NC50439题解

NC50439题解

题目描述
链接：https://ac.nowcoder.com/acm/problem/50439
来源：牛客网

在一个游戏中，tokitsukaze需要在n个士兵中选出一些士兵组成一个团去打副本。第i个士兵的战力为v[i]，团的战力是团内所有士兵的战力之和。
但是这些士兵有特殊的要求：如果选了第i个士兵，这个士兵希望团的人数不超过s[i]。(如果不选第i个士兵，就没有这个限制。)
tokitsukaze想知道，团的战力最大为多少。`

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100020;
#define LL long long
priority_queue<LL, vector<LL>, greater<LL> > q;
//形如“priority_queue<LL> q”是大根堆，上一行写法是小根堆，注意两个>之间一定要有一个空格
struct ty
{
    LL v;
    int s;
    bool operator < (const ty &a) const{
        return s > a.s;
    }
}a[maxn];
int main()
{
    int n;
    scanf("%d", &n);
    for(int i = 1; i <= n; i++)
    {
        scanf("%lld%d",&a[i].v, &a[i].s);
    }
    sort(a+1, a+n+1);
    //按照s从大到小排序，之后就根据这个数组来枚举队伍人数k
 
    LL tmp = 0, ans =0;
    //tmp用来存当前还在队伍里面的人的武力值之和，堆里的人就是还在队伍里的人
 
    for(int i = 1; i <= n; i++)
    {
        tmp += a[i].v;
        q.push(a[i].v);
        //把当前这个s最大的还没加入堆的人加入堆
 
        while(q.size() > a[i].s)
        {
            tmp -= q.top();
            q.pop();   //人数超出了就删除堆顶
        }
        //第i个人加入堆里之后，当前对人数的限制就是p[i].s(之前加入的人都比他的s值大)
        //这里就相当于题解说的从大到小枚举k
 
        ans = max(ans, tmp);
    }
    printf("%lld", ans);
    return 0;
}