FZU2219 StarCraft(哈夫曼树)

时间:2023-03-09 06:55:52
FZU2219 StarCraft(哈夫曼树)

一个工人可以变成两个工人,这样可以画出一颗二叉树,那么就是在叶子上建的建筑。

问题的时间花费,可以看作是这颗二叉树中各个叶子的深度*k+叶子对应建筑耗费时间中的最大值。

容易想到,类似哈夫曼树一样,从叶子出发往上合并,直到连通分量数小于等于m,结点的权值设定为w=max(lw,rw)+k。

 #include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
int main(){
int t,n,m,k,a;
scanf("%d",&t);
while(t--){
scanf("%d%d%d",&n,&m,&k);
priority_queue<int, vector<int>, greater<int> > que;
for(int i=;i<n;++i){
scanf("%d",&a);
que.push(a);
}
while(n>m){
int w1=que.top(); que.pop();
int w2=que.top(); que.pop();
que.push(w2+k);
--n;
}
int res=;
while(!que.empty()){
res=max(res,que.top());
que.pop();
}
printf("%d\n",res);
}
return ;
}