BZOJ[3728]PA2014 Final Zarowki 堆+贪心

题目链接http://www.lydsy.com/JudgeOnline/problem.php?id=3728

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。

题目大意：给定n个灯泡和n个房间，每个灯泡有一个功率，给出每个房间所需功率最小值，可以改变k个灯泡的功率值，每一个灯泡放到一个房间里，求最小的实际功率

可以考虑一种贪心的想法，对于每一个房间，选择功率比它大的最小功率的灯泡，如果没有则一定要换其他功率的灯泡
在这种想法上，我们可以想出一种贪心的放法将所有房间和灯泡功率从大到小排序，对于每一个房间，将比它大的灯泡扔入一个小根堆q中
如果堆为空则要换一个功率与其相等的灯泡，否则取出堆顶元素，将功率超过部分扔入大根堆q1中

如果最后灯泡还可以更换，则贪心更换q1堆顶的那个灯泡

注意答案要用long long储存

代码如下

#include<algorithm>
#include<cstdio>
#include<queue>
#define N 500050
using namespace std;
priority_queue<int,vector<int>,greater<int> >q;
priority_queue<int,vector<int>,less<int> >q1;
inline bool cmp(int a,int b){return a>b;}
int a[N],b[N];
int n,k,t,x;
long long ans;
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) scanf("%d",a+i);
    for(int i=1;i<=n;i++) scanf("%d",b+i);
    sort(a+1,a+n+1,cmp);sort(b+1,b+n+1,cmp);
    t=1;
    for(int i=1;i<=n;i++){
        while(a[t]>=b[i]&&t) q.push(a[t++]);
        if(q.empty()){
            ans=ans+b[i];
            if(k) {k--;continue;}
            return printf("NIE"),0;
        }
        x=q.top();q.pop();
        ans=ans+x;
        q1.push(x-b[i]);
    }
    while(k--&&!q1.empty()){ans=ans-q1.top();q1.pop();}
return printf("%lld",ans),0;
}