【bzoj3728】 PA2014Final Zarowki

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

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;
}