哈夫曼树 - 九度教程第30题
题目:
时间限制:1 秒 内存限制:32 兆 特殊判题:否
题目描述:
哈夫曼树,第一行输入一个数n,表示叶结点的个数。需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即weight,题目需要输出所有结点的值与权值的乘积之和。
输入:
输入有多组数据。
每组第一行输入一个数n,接着输入n个叶节点(叶节点权值不超过100,2<=n<=1000)。
输出:
输出权值。
样例输入:
5
1 2 2 5 9
样例输出:
37
来源:
2010年北京邮电大学计算机研究生机试真题
使用优先队列以后,求哈夫曼树的过程不仅时间复杂度降低许多(O(nlogn)),同时代码也轻便不少,所以使用数据结构堆来辅助求解哈夫曼树,是求哈夫曼树的最佳选择。在实际运用中确定哈夫曼树模型,最经典的问题模型为哈夫曼编码、多个数的两两合并等。
代码:
#include<stdio.h>
#include<queue> //标准模板库<queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> > Q;//建立一个小顶堆
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
while(Q.empty()==false) Q.pop(); //清堆中元素
for(int i=1;i<=n;i++) //输入n个叶子节点的权值
{
int x;
scanf("%d",&x);
Q.push(x); //将权值放入堆中
}
int ans=0; //保存答案
while(Q.size()>1) //当堆中元素大于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;
}