bzoj 3728: PA2014Final Zarowki

3728: 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。

Source

鸣谢Jcvb

题解：

贪心。。（话说贪心题怎么都有点难）

题目需要使用最小的功率，那么对于每一个可以被照亮的房间，应该对应最小功率的满足的灯泡。

排序后倒着枚举房间，此时将满足的灯泡加入堆，取最小元素。

若是堆为空，则需要换一次灯泡。

这题还有个注意点，就是灯泡也是可以去换小的，来使功率降低，所以在找到一对时，将差值放入另一个堆，最后次数多余的话更新ans

P.S.本蒟蒻表示不会stl。。。

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define p1 (p<<1)
#define p2 (p<<1|1)
const int N=500005;
int n,m,i,cnt,x,k,K,a[N],b[N],t[N],T[N];
long long ans;
inline void read(int &v){
    char ch,fu=0;
    for(ch='*'; (ch<'0'||ch>'9')&&ch!='-'; ch=getchar());
    if(ch=='-') fu=1, ch=getchar();
    for(v=0; ch>='0'&&ch<='9'; ch=getchar()) v=v*10+ch-'0';
    if(fu) v=-v;
}
void up(int p)
{
	while(p>>1)
	{
		if(t[p]>t[p>>1])
		{
			swap(t[p],t[p>>1]);
			p>>=1;
		} else break;
	}
}
void down(int p)
{
	int x;
	while(p1<=k)
	{
		if(p2<=k)
		{
			if(t[p1]>t[p2]) x=p1;else x=p2;
		} else x=p2;
		if(t[p]<t[x])
		{
			swap(t[p],t[x]);
			p=x;
		} else break;
	}
}
void Up(int p)
{
	while(p>>1)
	{
		if(T[p]<T[p>>1])
		{
			swap(T[p],T[p>>1]);
			p>>=1;
		} else break;
	}
}
void Down(int p)
{
	int x;
	while(p1<=K)
	{
		if(p2<=K)
		{
			if(T[p1]<T[p2]) x=p1;else x=p2;
		} else x=p2;
		if(T[p]>T[x])
		{
			swap(T[p],T[x]);
			p=x;
		} else break;
	}
}
int main()
{
	read(n),read(m);
	for(i=1;i<=n;i++) read(a[i]);
	for(i=1;i<=n;i++) read(b[i]);
	sort(a+1,a+n+1);
	sort(b+1,b+n+1);
	x=n;
	for(i=n;i>=1;i--)
	{
		for(;x&&a[x]>=b[i];x--) T[++K]=a[x],Up(K);
		if(K)
		{
			ans+=T[1];
			t[++k]=T[1]-b[i];
			up(k);
			T[1]=T[K--];
			Down(1);
		} else cnt++,ans+=b[i];
		if(cnt>m)
		{
			printf("NIE");
			return 0;
		}
	}
	for(i=cnt+1;i<=m;i++)
	{
		ans-=t[1];
		t[1]=t[k--];
		down(1);
	}
	cout<<ans;
	return 0;
}