二叉树--Huffman编码树--优先队列解法

超链接：Huffman编码树

总时间限制: 

1000ms 

内存限制: 

65535kB

描述

构造一个具有n个外部节点的扩充二叉树，每个外部节点Ki有一个Wi对应，作为该外部节点的权。使得这个扩充二叉树的叶节点带权外部路径长度总和最小：

                                     Min( W1 * L1 + W2 * L2 + W3 * L3 + … + Wn * Ln)

Wi:每个节点的权值。

Li:根节点到第i个外部叶子节点的距离。

编程计算最小外部路径长度总和。

输入

第一行输入一个整数t，代表测试数据的组数。
对于每组测试数据，第一行输入一个整数n，外部节点的个数。第二行输入n个整数，代表各个外部节点的权值。
2<=N<=100

输出

输出最小外部路径长度总和。

样例输入

2
3
1 2 3
4
1 1 3 5

样例输出

9
17

分析：直接利用优先队列水一下。。。

#include <cstdio>
#include <iostream>
#include <queue>
#include <cstring>
using namespace std;

int t,n;
priority_queue<int,vector<int>,greater<int>> q;

int main()
{
	cin >> t;
	while(t--) {
	    cin >> n;
	    while(n--) {
	        int a;
	        cin >> a;
	        q.push(a);
	    }
	    int sum = 0;
	    while(!q.empty()) {
	        int a = q.top();q.pop();
	        if(!q.empty()){
	        	int b = q.top();q.pop();
	        	sum += a+b;
	        	q.push(a+b);
	        }
	    }
	    cout << sum << endl;
	}
	return 0;
}