构建哈夫曼树及编码

Beetle___

已于 2023-11-06 21:23:35 修改

阅读量1.5k

点赞数 2

文章标签：数据结构算法

于 2023-11-06 21:21:41 首次发布

本文链接：https://blog.csdn.net/Beetle___/article/details/134255086

版权

第1关：构建哈夫曼树

任务描述

本关任务：构建哈夫曼树，从键盘读入字符个数n及这n个字符出现的频率即权值，构造带权路径最短的最优二叉树（哈夫曼树）。

编程要求

根据提示，在右侧编辑器补充代码，输出哈夫曼树的值。

测试说明

平台会对你编写的代码进行测试：

输入说明第一行字符的个数n；第二行输入这n个字符的权值，即n个整数。

测试输入： 8 5 29 7 8 14 23 3 11

输出说明第一行输出提示信息；第二行开始输出插入哈夫曼树元素的值，每个项数据元素的数据项用\t隔开。

预期输出：

代码如下

# include <stdio.h>
# include <iostream>
# include <string.h>
using namespace std;


typedef struct			//define structure HuffmanTree
{   int weight;
int parent,lchild,rchild;
}HTNode,*HuffmanTree;

typedef char **  HuffmanCode;

void Select(HuffmanTree HT,int i,int &s1,int &s2) ;//选出HT树到i为止，权值最小且parent为0的2个节点
void HuffmanTreeing(HuffmanTree &HT,int *w,int n);//构建哈夫曼树函数 

void output(HuffmanTree HT,int m);//输出哈夫曼树

int main()            		
{ 
	HuffmanTree HT;
	HuffmanCode HC;
	int n,i;
	int *w;
	scanf("%d",&n);
	w=(int *)malloc(n*sizeof(int));
  	for(i=0;i<n;i++)
	{
		scanf("%d",&w[i]);		
	}
	HuffmanTreeing( HT , w  ,n  );	
	cout<<"哈夫曼树:"<<endl;
	output(HT,2*n-1);	
	return 0;
}

void Select(HuffmanTree HT,int i,int &s1,int &s2)  
{ 	//选出HT树到i为止，权值最小且parent为0的2个节点
	//s1 is the least of HT[].weight
	//s2 is the second least of HT[].weight
	/********** Begin **********/ 
	int j;
    s1 = 0;
    s2 = 0;
    for (j = 1; j <= i; ++j)
    {
        if (HT[j].parent == 0)
        {
            if (s1 == 0 || HT[j].weight < HT[s1].weight)
            {
                s2 = s1;
                s1 = j;
            }
            else if (s2 == 0 || HT[j].weight < HT[s2].weight)
            {
                s2 = j;
            }
        }
    }
	/********** End **********/
} 


void HuffmanTreeing(HuffmanTree &HT,int *w,int n) //构建哈夫曼树函数
{ // w存放n个字符的权值(均>0)，构造赫夫曼树HT
	/********** Begin **********/
	int m=2*n-1;
    HT=(HuffmanTree)malloc((m+1)*sizeof(HTNode));
    for(int i=1;i<=m;i++)
    {
        HT[i].weight=0;
        HT[i].parent=0;
        HT[i].lchild=0;
        HT[i].rchild=0;
    }
    for(int i=1;i<=n;i++)
    {
        HT[i].weight=w[i-1];
    }
    for (int i = n + 1; i <= m; i++)
    {
        int s1, s2;
        Select(HT, i - 1, s1, s2);
        HT[s1].parent = i;
        HT[s2].parent = i;
        HT[i].lchild = s1;
        HT[i].rchild = s2;
        HT[i].weight = HT[s1].weight + HT[s2].weight;
    }
	/********** End **********/
}   


void output(HuffmanTree HT,int m)
{ //输出哈夫曼树
	for(int i=1;i<=m;++i)
	{   
		cout<<"HT["<<i<<"] ="<<HT[i].weight<<"\t"<<HT[i]. parent<<"\t";
		cout<<"\t" <<HT[i]. lchild <<"\t" <<HT[i]. rchild<<endl;
	}
}

第2关：根据哈夫曼树构建哈夫曼编码