Code Up 9.7问题 C: 哈夫曼树（优先队列实现）

问题 C: 哈夫曼树

 

题目描述

哈夫曼树，第一行输入一个数n，表示叶结点的个数。需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出所有结点的值与权值的乘积之和。

输入

输入有多组数据。
每组第一行输入一个数n，接着输入n个叶节点（叶节点权值不超过100，2<=n<=1000）。

输出

输出权值。

样例输入

2
2 8 
3
5 11 30 

样例输出

10
62

参考代码：

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

int n;
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		//代表小顶堆的优先队列
		priority_queue<long long ,vector<long long>,greater<long long> >q;
		int a,b,ans=0;
		for(int i=0;i<n;++i)
		{
			scanf("%d",&a);
			q.push(a);
		}
		while(q.size()>1)//只要优先队列中至少有两个元素 
		{
			a=q.top();
			q.pop();
			b=q.top();
			q.pop();
			ans+=a+b;
			q.push(a+b);
		}
		printf("%d\n",ans);
	}
	
	return 0;
}