题目大意:
题目链接:https://ac.nowcoder.com/acm/contest/1080/C
在一个游戏中,tokitsukaze需要在
n
n
n个士兵中选出一些士兵组成一个团去打副本。
第
i
i
i个士兵的战力为
v
[
i
]
v[i]
v[i],团的战力是团内所有士兵的战力之和。
但是这些士兵有特殊的要求:如果选了第
i
i
i个士兵,这个士兵希望团的人数不超过
s
[
i
]
s[i]
s[i]。(如果不选第i个士兵,就没有这个限制。)
tokitsukaze想知道,团的战力最大为多少。
思路:
显然的,先按
s
s
s从小到大排序,这样我们就可以枚举
s
s
s最小的士兵
i
i
i,然后其他不超过
s
[
i
]
s[i]
s[i]个士兵必须全部在
[
i
+
1
,
n
]
[i+1,n]
[i+1,n]中选择。
也就是说,我们需要在
O
(
log
n
)
O(\log n)
O(logn)维护出区间
[
l
,
r
]
[l,r]
[l,r]的前
k
k
k大的和。
所以我们就从右往左枚举,每次把
v
[
i
]
v[i]
v[i]插入一个大根堆中,同时维护另一个小根堆,小根堆内维护选择的不超过
s
[
i
]
s[i]
s[i]个士兵的权值。在区间
[
i
,
n
]
[i,n]
[i,n]中没有选择的士兵权值就放在另一个大根堆中。
往前枚举一位后,如果
s
[
i
]
s[i]
s[i]更大,那么就从小根堆中弹出元素到大根堆中,知道元素个数等于
s
[
i
]
s[i]
s[i]。如果
s
[
i
]
s[i]
s[i]更小,那么我们就从大根堆内不断弹出元素插入小根堆中,知道小根堆大小等于
s
[
i
]
s[i]
s[i]。
维护好小根堆的元素个数等于
s
[
i
]
s[i]
s[i]后,我们就不断比较大根堆的堆顶是否大于小根堆的堆顶,如果大于就交换两者,保证小根堆内元素和一定最大。
W
Y
C
d
a
l
a
o
o
r
z
WYCdalao\ orz
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;
}