Input
第一行包含一个整数T(T<=50),表示数据组数。
每组数据第一行包含一个整数n(2<=n<=1000),表示果子的堆数。
第二行包含n个正整数ai(ai<=100),表示每堆果子的果子数。
Output
每组数据仅一行,表示最小合并代价。
Sample Input
2
4
1 2 3 4
5
3 5 2 1 4
Sample Output
19
33
我们有时不需要真的去构建一棵哈夫曼树,只需要能得到最终的带权路径长度即可,(例如合并果子问题就是只需要知道消耗的最少体力即可),我们只需要掌握哈夫曼的构
现在有n堆果子,第i堆有ai个果子。现在要把这些果子合并成一堆,每次合并的代价是两堆果子的总果子数。求合并所有果子的最小代价。建思想就行了,也就是反复选择2个最小的元素,合并直到只队列剩下一个元素。
#include<iostream>
#include<queue>
using namespace std;
priority_queue<long long, vector<long long>, greater<long long> > q;
int main (){
int T;
scanf("%d", &T);
while(T--){
int n;
long long temp, x, y, ans = 0;
while(!q.empty()) q.pop();
scanf("%d", &n);
for(int i = 0; i < n; i++){
scanf("%lld", &temp);
q.push(temp);
}
while(q.size() > 1){ //只要优先队列中至少有两个元素
x = q.top();
q.pop();
y = q.top();
q.pop();
q.push(x + y);
ans += x + y;
}
printf("%lld\n", ans);
}
return 0;
}