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

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;
} 

 

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值