哈夫曼树(优先队列)合并果子

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
#include 
using namespace std;

priority_queue, greater > 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;
} 

 

你可能感兴趣的:(树,堆,~~~~~STL容器~~~~~,~~~~~数据结构~~~~~)