BZOJ[3728]PA2014 Final Zarowki

/*
将手头上有的亮度a及要求的房间亮度b都降序排出来
一个个拿出要求的亮度，将大于等于它的a的值一个个丢到小根堆q里
然后看下q是否为空，如果为空说明此时要去商店拿一个亮度与当前一致的
如果q不为空，则弹出堆顶值，并将两者的差值加到一个大堆根q1中
全部做完后，如果还有名额去商店拿灯泡，则弹出q1的堆顶 
*/
#include<bits/stdc++.h>
#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) //将大于第i个所要求的灯亮度的灯丢到小根堆里 
             q.push(a[t++]);
        if(q.empty()) //如果堆为空 
        {
            ans=ans+b[i];//使用当前亮度 
            if(k) //如果还可以换灯泡 
                    k--;
            else //堆中没有大于当前要求亮度的，又没办法去商店拿，于是无解 
                   return printf("NIE"),0;
        }
        else //说明堆中还有元素值大于当前亮度的 
        {
           x=q.top();
           q.pop();
           ans=ans+x;//从小根堆中取出一个大于等于当前要求亮度的且值最小的，用上之 
           q1.push(x-b[i]);//差值放到大根堆中 
        }
     }
    while(k--&&!q1.empty())//如果还可以换灯泡的话 
    {
        ans=ans-q1.top();
    //    cout<<"change it ans is "<<ans<<endl;
        q1.pop();
    }
return printf("%lld",ans),0;
}