【bzoj3728】PA2014Final Zarowki

Description

有n个房间和n盏灯，你需要在每个房间里放入一盏灯。每盏灯都有一定功率，每间房间都需要不少于一定功率的灯泡才可以完全照亮。
你可以去附近的商店换新灯泡，商店里所有正整数功率的灯泡都有售。但由于背包空间有限，你至多只能换k个灯泡。
你需要找到一个合理的方案使得每个房间都被完全照亮，并在这个前提下使得总功率尽可能小。

Input

第一行两个整数n,k(1<=k<=n<=500000)。
第二行n个整数pi，表示你现有的灯泡的功率。
第三行n个整数wi，表示照亮每间房间所需要的最小功率。

Output

如果无法照亮每间房间，仅输出NIE。
否则输出最小的总功率。

Sample Input

6 2

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。

题解
p，w都从大到小排序，枚举w，将大于等于w[i]的p加入小根堆，如果堆中有数，则取堆顶灯泡，再将p-w加入另一个大根堆（因为如果k剩余可以将大根堆灯泡功率变小），否则k-1。

代码

#include<bits/stdc++.h>
#define mod 1000000007
#define inf 10000000
#define N 1000105
#define pa pair<long long,int>
typedef long long ll;
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,k,p[500005],w[500005];
ll ans;
priority_queue<int,vector<int>,greater<int> >mn;
priority_queue<int,vector<int> >mx;
int main()
{
    n=read();k=read();
    for (int i=1;i<=n;i++) p[i]=read();sort(p+1,p+n+1,greater<int>());
    for (int i=1;i<=n;i++) w[i]=read();sort(w+1,w+n+1,greater<int>());
    int l=1;
    for (int i=1;i<=n;i++)
    {
        while (p[l]>=w[i]&&l<=n) mn.push(p[l++]);
        if (mn.empty()) k--,ans+=w[i];
        else ans+=mn.top(),mx.push(mn.top()-w[i]),mn.pop();
    }
    if (k<0) return printf("NIE"),0;
    while (k--&&!mx.empty())
    {
        ans-=mx.top();
        mx.pop();
    }
    printf("%lld",ans);
    return 0;
}