【牛客练习赛50】C - tokitsukaze and Soldier【对顶堆】

题目大意：

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

---

思路：

显然的，先按$s$从小到大排序，这样我们就可以枚举$s$最小的士兵$i$，然后其他不超过$s[i]$个士兵必须全部在$[i+1,n]$中选择。
也就是说，我们需要在$O(\log n)$维护出区间$[l,r]$的前$k$大的和。
所以我们就从右往左枚举，每次把$v[i]$插入一个大根堆中，同时维护另一个小根堆，小根堆内维护选择的不超过$s[i]$个士兵的权值。在区间$[i,n]$中没有选择的士兵权值就放在另一个大根堆中。
往前枚举一位后，如果$s[i]$更大，那么就从小根堆中弹出元素到大根堆中，知道元素个数等于$s[i]$。如果$s[i]$更小，那么我们就从大根堆内不断弹出元素插入小根堆中，知道小根堆大小等于$s[i]$。
维护好小根堆的元素个数等于$s[i]$后，我们就不断比较大根堆的堆顶是否大于小根堆的堆顶，如果大于就交换两者，保证小根堆内元素和一定最大。
$WYCdalao\ orz$一个堆就可以搞定本题。
对啊为什么我要用对顶堆啊

---

代码：

#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;
typedef long long ll;

const ll N=100010;
priority_queue<ll> q,p;
ll ans,maxn,n;

struct node
{
	ll v,s;
}a[N];

bool cmp(node x,node y)
{
	return x.s<y.s;
}

int main()
{
	scanf("%lld",&n);
	for (ll i=1;i<=n;i++)
		scanf("%lld%lld",&a[i].v,&a[i].s);
	sort(a+1,a+1+n,cmp);
	for (ll i=n;i>=1;i--)
	{
		q.push(a[i].v);
		while (q.size() && p.size()<a[i].s)
		{
			ans+=(ll)q.top();
			p.push(-q.top());
			q.pop();
		}
		while (p.size()>a[i].s)
		{
			ans-=(-p.top());
			q.push(-p.top());
			p.pop();
		}
		while (q.size() && p.size() && q.top()>-p.top())
		{
			ans-=(-p.top()); ans+=q.top();
			q.push(-p.top()); p.pop();
			p.push(-q.top()); q.pop();
		}
		maxn=max(maxn,ans);
	}
	printf("%lld",maxn);
	return 0;
}