题目1106: 哈夫曼树

题目描述

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

 

输入

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

 

输出

输出权值。

 

样例输入

2

2 8 

3

5 11 30 

 

样例输出

10

62

 

提示 [+]

*** 提示已隐藏，点击上方 [+] 可显示 ***

 

来源

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

 

/********************************* 
 *    日期：2013-3-15
 *    作者：SJF0115 
 *    题号: 天勤OJ 题目1106: 哈夫曼树
 *    来源：http://www.acmclub.com/problem.php?id=1106
 *    结果：AC 
 *    来源：2010年北京邮电大学计算机研究生机试真题
 *    总结： 
**********************************/
#include<iostream>
#include<stdio.h>
#include<queue>
using namespace std;

int main(){
	int i,N,x,Weight;
    while(scanf("%d",&N) != EOF){
		Weight = 0;
		//建立一个小顶堆
		priority_queue<int, vector<int>, greater<int> > MinHeap;
		//输入n个叶子节点的权值
		for(i = 0;i < N;i++){
			scanf("%d",&x);
			MinHeap.push(x);
		}
		//当堆中元素大于1
		while(MinHeap.size() > 1){
			//取出堆中最小的两个元素，他们同一个结点的左右儿子，且双亲结点的权值为他们的和
			int a = MinHeap.top();
			MinHeap.pop();
			int b = MinHeap.top();
			MinHeap.pop();
			//权值和
			Weight += a + b;
			MinHeap.push(a + b);
		}
		printf("%d\n",Weight);
	}//while
	return 0;
}

此题另一解：点击打开链接