牛客 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

题目分析

1. 首先要注意题目给的范围，所以我们要用long long 去定义武力值
2. 在排序的时候我们让要求人数多的士兵排在前头
3. 接下来我们就一次枚举每一种情况，最后得出最大值，我们用一个优先队列去存入每一个士兵的武力值，并且在每一次存入的时候都去判断是否满足当前士兵的要求（最多s[i]个人），如果超出我们就除去栈顶的元素，因为我们定义的是小根堆，排序规则为从小到大，栈顶的元素肯定是最小的，这样我们就能兼顾每一位士兵的要求，从而求出武力值最大的团队去打副本

AC代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 10;
typedef long long ll;
struct node
{
   
   ll v;
   int s;
   bool operator<(const node x) const
   {
   
      return s > x.s; //结构体排序让s[i]多的在前面
   }
} w[maxn];
priority_queue<ll, vector<ll>, greater<ll> >q; //小根堆
//priority_queue<ll>q 默认是大根堆
int main()
{
   
   int n;
   while (~scanf("%d", &n))
   {
   
      for (int i = 1; i <= n; ++i)
         scanf("%lld%d", &w[i].v, &w[i].s);
      sort(w + 1, w + n + 1); //排序方便后续计算
      ll cnt = 0, ans = 0;
      for (int i = 1; i <= n; ++i)
      {
   
         cnt += w[i].v;
         q.push(w[i].v);           //用堆去维护武力值
         while (q.size() > w[i].s) //每次加入的时候都判断是否满足当前士兵的要求
         {
   
            cnt -= q.top();
            q.pop();
         }
         ans = max(ans, cnt); //计算最值
      }
      printf("%lld\n", ans);
   }
   return 0;
}