「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
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中最大的，将p-w放入大根堆
未匹配的w使用k
若k还有剩余，从堆中取k个更新答案
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
#include<iostream> #include<cstring> #include<cstdio> #include<cstdlib> #include<cmath> #include<algorithm> #include<vector> #include<set> #include<map> #include<queue> #define ll long long #define inf 1000000000 using namespace std; ll read() { ll 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; } ll ans; int n,k; int p[500005],w[500005]; multiset<int> st; priority_queue<int,vector<int> >q; int main() { n=read();k=read(); for(int i=1;i<=n;i++)p[i]=read(); sort(p+1,p+n+1); for(int i=1;i<=n;i++)w[i]=read(); for(int i=1;i<=n;i++) st.insert(w[i]); for(int i=1;i<=n;i++) { multiset<int>::iterator it=st.upper_bound(p[i]); if(it!=st.begin()) { --it; q.push(p[i]-*it); st.erase(it); ans+=p[i]; } } if(st.size()>k){puts("NIE");return 0;} for(multiset<int>::iterator it=st.begin();it!=st.end();it++) { k--; ans+=*it; } while(k--){ans-=q.top();q.pop();} printf("%lld\n",ans); return 0; } |
Subscribe
计数器 9787734
OI 课件=>
OI 课件=>