九度OJ题目1172：哈夫曼树

题目1172：哈夫曼树

时间限制：1 秒

内存限制：32 兆

特殊判题：否

题目描述：

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

输入：

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

输出：

输出权值。

样例输入：

5  
1 2 2 5 9

样例输出：

37

来源：

2010年北京邮电大学计算机研究生机试真题

#include <stdio.h>
#include <queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> > q;//定义小顶堆q 
int main(){
	int n;
	while(scanf("%d",&n)!=EOF){
		while(q.empty() == false)  q.pop();//清空堆中元素
		for(int i=0;i<n;i++){ //输入n个叶子结点权值
			int x;
			scanf("%d",&x);
			q.push(x); //将权值放入堆中
		}
		int ans=0;//保存答案
		while(q.size()>1){
			int a=q.top();
			q.pop();
			int b=q.top();
			q.pop();//取出堆中两个最小元素
			ans += a+b;//累加权值
			q.push(a+b);//将该双亲节点的权值放回堆中
		}
		printf("%d\n",ans);//输出答案
	}
	return 0;
}
/************************************************************** 
    Problem: 1172 
    User: zpy
    Language: C++ 
    Result: Accepted 
    Time:0 ms 
    Memory:1024 kb 
****************************************************************/