[PA2014]Żarówki

[PA2014]Żarówki

题目大意：

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

思路：

贪心，将房间按\(w_i\)从大到小排序。找到最小的能满足该房间的灯。如果不存在，就花一次换灯泡的机会，换来一个功率恰好等于\(w_i\)的灯泡。最后如果还有多的机会，就将浪费功率最多的灯泡换掉。

源代码：

#include<set>
#include<cstdio>
#include<cctype>
#include<algorithm>
#include<functional>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x; 
}
typedef long long int64;
const int N=5e5+1;
std::multiset<int> p,t;
int w[N];
int main() {
    const int n=getint();
    int k=getint();
    for(register int i=1;i<=n;i++) {
        p.insert(getint());
    }
    for(register int i=1;i<=n;i++) {
        w[i]=getint();
    }
    std::sort(&w[1],&w[n]+1,std::greater<int>());
    int64 ans=0;
    for(register int i=1;i<=n;i++) {
        std::multiset<int>::iterator it=p.lower_bound(w[i]);
        if(it==p.end()) {
            if(!k--) {
                puts("NIE");
                return 0;
            }
            ans+=w[i];
        } else {
            ans+=*it;
            if(*it-w[i]) {
                t.insert(*it-w[i]);
            } 
            p.erase(it);
        }
    }
    while(k--&&!t.empty()) {
        ans-=*t.rbegin();
        t.erase(--t.end());
    }
    printf("%lld\n",ans);
    return 0;
}