[牛客每日一题] (贪心+小顶堆)NC50439 tokitsukaze

Posted 2022-01-13 泥烟

tokitsukaze and Soldier (nowcoder.com)https://ac.nowcoder.com/acm/problem/50439

题目描述

在一个游戏中，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], 就去除战力最小的

即在满足一部分人的需求的情况下, 去除多余的"弱者"

比较好维护且每次可以对最值进行计算/删除的数据结构, 那么就可以选择堆

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;

const int N = 1e5+10;
struct solider
	int v;
	int s;
	bool operator < (const solider& y)
		return s > y.s;
	
so[N]; 

int main()

	priority_queue<int, vector<int>, greater<int> > q;//小根堆 
	
	int n;
	cin >> n;
	
	for(int i = 0; i < n ; i ++)
		scanf("%d%d", &so[i].v, &so[i].s);
	
	sort(so, so+n);
	long long res = 0, tmp = 0;
	
	for(int i = 0; i < n; i ++)
	
		tmp += so[i].v;
		q.push(so[i].v);
		
		while(q.size() > so[i].s)
		
			tmp -= q.top();
			q.pop();
		
		
		res = max(res, tmp);
	
	cout << res;
	return 0;