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
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
题解:
贪心。。(话说贪心题怎么都有点难)
题目需要使用最小的功率,那么对于每一个可以被照亮的房间,应该对应最小功率的满足的灯泡。
排序后倒着枚举房间,此时将满足的灯泡加入堆,取最小元素。
若是堆为空,则需要换一次灯泡。
这题还有个注意点,就是灯泡也是可以去换小的,来使功率降低,所以在找到一对时,将差值放入另一个堆,最后次数多余的话更新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;
}