百练 4080:Huffman编码树

4080:Huffman编码树
查看 提交 统计 提示 提问
总时间限制: 1000ms 内存限制: 65536kB
描述
构造一个具有n个外部节点的扩充二叉树，每个外部节点Ki有一个Wi对应，作为该外部节点的权。使得这个扩充二叉树的叶节点带权外部路径长度总和最小：
Min( W1 * L1 + W2 * L2 + W3 * L3 + … + Wn * Ln)
Wi:每个节点的权值。
Li:根节点到第i个外部叶子节点的距离。
编程计算最小外部路径长度总和。
输入
第一行输入一个整数n，外部节点的个数。第二行输入n个整数，代表各个外部节点的权值。
2<=N<=100
输出
输出最小外部路径长度总和。
样例输入
4
1 1 3 5
样例输出

17

这题很水，秒过

# include <iostream>
# include <queue>
using namespace std;
priority_queue<int, vector<int>, greater<int> > q;
int main(){
	int n, num, num1, num2, i;
	while(cin>>n){
		while(!q.empty()){
			q.pop();
		}
		for(i=1; i<=n; i++){
			cin>>num;
			q.push(num);
		}
		int ans=0;
		while(q.size()!=1){
			num1=q.top();
			q.pop();
			num2=q.top();
			q.pop();
			ans=ans+num1+num2;
			q.push(num1+num2);
		}
		cout<<ans<<endl;
	}
	return 0;
}