问题 B: 哈夫曼树 [priority_queue]

题目描述

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

输入

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

输出

输出权值。

样例输入

2
2 8 
3
5 11 30 

样例输出

10
62

-------------------------------------------------------------------------------

-----------------------------------------------------------------------------------------

即为除叶节点外所有权值之和 

#include<iostream>
#include<queue>
#include<vector>//要使用向量，因为使用了优先级队列
using namespace std;
int main()
{
	int n;
	while(scanf("%d",&n)==1)//scanf("%d",&n)!=EOF也OK
	{
		int w=0;
------------------------------------------------------------------
		priority_queue<int,vector<int>,greater<int> >q;//默认从大到小，加greater以后从                    小到大，top到back的顺序

大-->小 top-->back

------------------------------------------------------------------
		for(int i=0;i<n;i++)
		{
			int v;
			scanf("%d",&v);
			q.push(v); 
		}
		while(q.size()>1)//不能用q.empty()，q中定会保留一个根节点的值	
    	{
			int a=q.top();
			q.pop();
			int b=q.top();
			q.pop();
			w+=a+b;
			q.push(a+b); 
		}
		printf("%d\n",w);
	}
	return 0;
}