Huffuman树
Description
Huffman树在编码中有着广泛的应用。在这里,我们只关心Huffman树的构造过程。
给出一列数 { pi } = { p0, p1, …, pn-1 },用这列数构造Huffman树的过程如下:
- 找到{pi}中最小的两个数,设为 pa 和 pb,将 pa 和 pb 从 { pi } 中删除掉,然后将它们的和加入到 { pi } 中。这个过程的费用记为pa + pb。
- 重复步骤1,直到 { pi } 中只剩下一个数。
在上面的操作过程中,把所有的费用相加,就得到了构造Huffman树的总费用。
本题任务:对于给定的一个数列,现在请你求出用该数列构造Huffman树的总费用。
例如,对于数列 { pi } = { 5, 3, 8, 2, 9 },Huffman树的构造过程如下: - 找到{5, 3, 8, 2, 9}中最小的两个数,分别是2和3,从{pi}中删除它们并将和5加入,得到{5, 8, 9, 5},费用为5。
- 找到{5, 8, 9, 5}中最小的两个数,分别是5和5,从{pi}中删除它们并将和10加入,得到{8, 9, 10},费用为10。
- 找到{8, 9, 10}中最小的两个数,分别是8和9,从{pi}中删除它们并将和17加入,得到{10, 17},费用为17。
- 找到{10, 17}中最小的两个数,分别是10和17,从{pi}中删除它们并将和27加入,得到{27},费用为27。
- 现在,数列中只剩下一个数27,构造过程结束,总费用为5+10+17+27=59。
Input
多测试用例,每个测试用例的第一行是一个正整数 n( n ≤ 100000 )。
接下来是n个正整数,表示p0, p1, …, pn-1,每个数不超过1000。
Output
每个测试用例输出一行:用这些数构造Huffman树的总费用。
Sample Input
5
5 3 8 2 9
Sample Output
59
Author
John
相信大家看题目都知道怎么建哈夫曼树的,如果还不懂也没关系,点这里(图文详解不懂都不行,写的太好了)https://blog.csdn.net/qq_33990383/article/details/53073825
现在双手呈上本题代码:
#include<bits/stdc++.h>
using namespace std;
priority_queue<int,vector<int>, greater <int> > q;//小顶堆 升序
int main(void)
{
int sum,n,a,b;
while(~scanf("%d",&n))
{
while(!q.empty())
{
q.pop();
}
for(int i=1;i<=n;i++){
scanf("%d",&a);
q.push(a);
}
sum=0;
//建立哈夫曼树
while(q.size()>1)
{
a=q.top();q.pop();//取出最小的数(子树的权重)
sum+=a;
b=q.top();q.pop();//再取出最小的数(子数的权重)
sum+=b;
q.push(a+b);
}//也是计算最优二叉树的权值
printf("%d\n",sum);
}
return 0;
}
代码很短,我们使用到了优先队列(优先队列可以理解为堆,好像说堆是优先队列的一种实现方式,就是一种可以很快拿出最小值的数据结构)来建立(说是建立哈夫曼树,其实个人理解是计算而不是建立,为什么是“计算”呢,因为最后优先队列q只剩下一个点,但这个过程充分体现并利用了哈夫曼树的建立过程)哈夫曼树。