Description
有n个房间和n盏灯,你需要在每个房间里放入一盏灯。每盏灯都有一定功率,每间房间都需要不少于一定功率的灯泡才可以完全照亮。
你可以去附近的商店换新灯泡,商店里所有正整数功率的灯泡都有售。但由于背包空间有限,你至多只能换k个灯泡。
你需要找到一个合理的方案使得每个房间都被完全照亮,并在这个前提下使得总功率尽可能小。
Input
第一行两个整数n,k(1<=k<=n<=500000)。
第二行n个整数p[i](1<=p[i]<=10^9),表示你现有的灯泡的功率。
第三行n个整数w[i](1<=w[i]<=10^9),表示照亮每间房间所需要的最小功率。
Output
如果无法照亮每间房间,仅输出NIE。
否则输出最小的总功率。
Sample Input
6 2
12 1 7 5 2 10
1 4 11 4 7 5
12 1 7 5 2 10
1 4 11 4 7 5
Sample Output
33
HINT
解释:将2和10换成4和4。配对方案为1-1,4-4,4-4,5-5,7-7,11-12。
Solve
先排序,用两个堆维护,一个小根堆存大于等于当前功率的灯,另一个大根堆存已配对的灯与该房间功率的差值
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<queue>
#define int long long
using namespace std;
priority_queue<int>p;
priority_queue< int,vector<int>,greater<int> >q;
int n,m,ans,a[500005],b[500005],tmp;
main (){
scanf ("%d%d",&n,&m);
for (int i=1;i<=n;++i)scanf ("%lld",&a[i]);
for (int i=1;i<=n;++i)scanf ("%lld",&b[i]);
sort(a+1,a+n+1);sort(b+1,b+n+1);
for (int i=n,j=n;i;--i){
while(a[j]>=b[i] && j)q.push(a[j--]);
if (q.empty()){
if (!m){
puts("NIE");
return 0;
}
ans+=b[i];
m--;
}
else{
tmp=q.top();
q.pop();
ans+=tmp;
p.push(tmp-b[i]);
}
}
while (m--){
tmp=p.top();
ans-=tmp;
p.pop();
}
printf ("%lld",ans);
return 0;
}