构建HaffmanTree
上课的时候老师讲到HaffmanTree的时候,看了下书上的伪代码了解了大致的思路就开始了在自己电脑上代码实现,不过搞笑的是想着下课就给实现了,没想到铃声一响运行建树结果不对。
二叉树基本采用链式存储,但我们这里使用顺序存储来实现:
书上给出的伪代码是这样的:
基本算法思想是:
每次搜索当前所有未合并结点中最小的两个,把这两个结点的parent赋值为新建节点的下标,新建节点的lchild和rchild赋值为这两个最小的结点下标,新建结点的权值赋为两个结点权值之和
下面给出一个HaffmanTree建树例题:
根据算法思想可以手绘HaffmanTree树,并算出其带权路径长度:
这里大家可能会很疑惑n=8,m=15是什么呀?
当然n是给定的结点数,那m呢。这里我们知道HaffmanTree是一颗二叉树,没有度为1的结点,即就是n0 = n , n1 = 0 , 则n2(度为2的结点数) = n-1。
可能很多小伙伴又有疑惑了,这结论怎么推导出来的?
是这样,在二叉树中,度为0的结点数等于度为2的结点数加1(n0 = n2 + 1)
这些问题都搞清楚之后,下面给出源代码:
#include <iostream>
#include <algorithm>
using namespace std;
typedef struct{
int weigth;
int parent,lchild,rchild;
}HTNode,*HaffmanTree;
//找到HaffmanTree中,parent为0的节点,找出两个最小值,把下标赋值给s1和s2
//每次查找的范围需要变化,因为新加入的节点的parent的值仍为0
//这里parent值为0表示该结点没有被合并
void Select(HaffmanTree HT,int n,int &s1,int &s2)
{
int a[n];
int count = 0;
for(int i = 1 ; i <= n ; i++){
if(HT[i].parent == 0){
a[count] = HT[i].weigth;
count++;
}
}
sort(a,a+count);//默认从小到大排序
//找出最小的两个结点
for(int i = 1 ; i <= n ; i++){
if(a[0] == HT[i].weigth && HT[i].parent == 0){
s1 = i;
break;
}
}
for(int i = 1 ; i <= n ; i++){
//下面这一步很重要:要保证第二小的结点和上面找的最小的结点不是一个结点(可能会出现两个结点的权值相同的情况)
if(i != s1 && a[1] == HT[i].weigth && HT[i].parent == 0){
s2 = i;
break;
}
}
}
void CreatHaffmanTree(HaffmanTree &HT,int n)
{
if(n <= 1)
return ;
int m = 2 * n - 1;//总结点数
HT = new HTNode[m + 1];//HT指向一段长为m+1的数组
for(int i = 0 ; i <= m ; i ++){
HT[i].lchild = 0;
HT[i].parent = 0;
HT[i].rchild = 0;
}
//初始化1-n,即就是前n个已经给点节点的权值
for(int i = 1; i <= n ;i ++){
cin >> HT[i].weigth;
}
//开始创建HaffmanTree
for(int i = n + 1 ; i <= m ; i ++){
int s1 = 0,s2 = 0;
Select(HT , i - 1 , s1 , s2);
//删除当前最小的两个节点,即就是把节点的parent值赋值
HT[s1].parent = i;
HT[s2].parent = i;
HT[i].lchild = s1;
HT[i].rchild = s2;
HT[i].weigth = HT[s1].weigth + HT[s2].weigth;
}
for(int i = 1; i <= m ; i ++){
printf("%3d%3d%3d%3d%3d\n",i , HT[i].weigth , HT[i].parent , HT[i].lchild , HT[i].rchild );
}
}
int main()
{
HaffmanTree HT;
int n;//节点数
cin>>n;
CreatHaffmanTree(HT,n);
return 0;
}
关于代码中,查找两个最小权值结点部分,开始我的想法是在一个for循环里面实现,但是最后没有成功,如果读者有更好的算法实现,请多指教,私信交流哈!
运行结果:
对比一下书上的初态以及终态表格,没有错误