实验5.1 统计字符种类及其出现的频率创建哈夫曼树

头文件

#include<dos.h>
#include<conio.h>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>

定义结构体

typedef struct//定义哈夫曼树
{
	unsigned int weight; //结点权值
	unsigned int parent, lchild, rchild; //结点的双亲，左，右指针
}HTNode,*HuffmanTree; 

typedef char** HuffmanCode; //动态分配存储哈夫曼编码的空间 
// char **为二级指针，保存一级指针 char *的地址，

在HT【1……t】中选择权值最小结点

int minpoint(HuffmanTree HT, int n)
{
	int i; 
	int min;        //用来存放weight最小且parent为0的元素的序号  
	int min_weight=32767; //用来存放weight最小且parent为0的元素的weight值   
	//选出weight最小且parent为0的元素，并将其序号赋给min  
	for (i = 1; i <= n; i++)
	{
		if (HT[i].weight < min_weight && HT[i].parent == 0)
		{
			min_weight = HT[i].weight;
			min = i;
		}
	}
	//选出weight最小的元素后，将其parent置1，使得下一次比较时将其排除在外。  
	HT[min].parent = 1;
	return min;
}

从HT数组的前k个元素中选出weight最小且parent为0的元素，并将该元素的序号返回

void select(HuffmanTree HT, int n, int &s1, int &s2)
 {
	s1 = minpoint(HT, n);
	s2 = minpoint(HT, n);
 }

构造一棵赫夫曼树

void CreatHaffmantree(HuffmanTree& HT, unsigned int* w, int n)
{
	//w存放n个结点的权值
	int i, m; //i用于for循环，m表示哈夫曼树的总结点数
	int s1, s2; //s1，s2表示权值最小的两个结点的编号
	HuffmanTree p;
	if (n <= 1) // 判断输入是否合规 
		return;//当结点数不满足创建哈夫曼树的条件时，直接退出本函数
	m = 2 * n - 1; //叶子结点数为n时的总结点数
	/*设叶子节点个数为n，度为1的节点个数为m，度为2的节点个数为p.
	由 p + m + n = 2p + m + 1得n=p+1 故哈夫曼树的总结点=n+n+1=2*n+1*/
	HT = (HuffmanTree)malloc((m + 1) * sizeof(HTNode));//开辟2*n个结点空间，0号单元不用
	if (HT == NULL)
	{
		printf("\n分配内存失败！\n");
		return;
	}
	// 进行初始化
	//先对已知权值的结点进行初始化
	p = HT + 1;//0号单元不用 
	for (i = 1;i <= n;++i, ++p, ++w)
	{
		p->weight = *w; //函数列表中的参数w为指针，*w才表示指针指向的数值
		//初始时，p的双亲孩子都置0
		p->parent = 0;
		p->lchild = 0;
		p->rchild = 0;
	}
	//再对未知权值的结点进行初始化
	for (;i <= m;++i, ++p)
	{
		p->weight = 0;
		p->lchild = 0;
		p->rchild = 0;
		p->parent = 0;
	}
	printf("\n初始化完成\n");
	//初始化结束，开始构建哈夫曼树
	for (i = n + 1;i <= m;i++)
	{
		select(HT, i - 1, s1, s2); //选择两双亲结点为0且权值最小的结点 
		//构造一棵新的子树
		//修改s1s2的双亲结点
		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;
	}
	printf("\n创建哈夫曼树结束!\n");
}

将有n个叶子节点的哈夫曼树编码，将所编码存放于HC中

void huffmancoding(HuffmanTree HT, HuffmanCode& HC, int n)
{
	//从叶子到根逆向求每个叶子结点的哈夫曼编码
	int i, c, f, start;
	//i用于for循环 c指示结点的编号 f表示c的双亲的结点编号 start表示编码数组的下标
	char* cd;
	//分配n个编码的头指针向量
	HC = (HuffmanCode)malloc((n + 1) * sizeof(char*));
	//开辟一个求编码的工作空间
	cd = (char*)malloc(n * sizeof(char));
	cd[n - 1] = '\0'; //编码结束符
	for (i = 1;i <= n;++i) //逐个叶子节点
	{
		//n个结点有n-1条边
		start = n - 1;//因为是逆向，所以其实为n-1
		//从叶子到根逆序求编码
		for (c =i, f = HT[i].parent;f != 0;c = f, f = HT[f].parent) 
		{
			if (HT[f].lchild == c)
				cd[--start] = '0';//左孩子的编码为0
			else
				cd[--start] = '1'; //右孩子的编码为1
		}
		HC[i] = (char*)malloc((n - start) * sizeof(char)); //为第i个编码分配空间
		strcpy(HC[i], &cd[start]); //将编码从cd复制到HC中
		//strcpy只用于字符串复制，遇到‘\0’时停止，还会复制字符串的结束符'\0'；
	}
	free(cd); //释放临时创建的空间
}

输出该哈夫曼树编码表

void printhuffmantree(HuffmanCode HC, unsigned int* w, int n)
{
	int i;
	printf("\nHuffmanCode is:\n");
	for (i = 1;i <= n;i++)
	{
		//输出数字及其对应的哈夫曼编码
		printf("%3d---", w[i - 1]); 
		puts(HC[i]);
		//把字符串输出到标准输出设备，将'\0'转换为回车换行。
	}
	printf("\n");
	printf("\n完成哈夫曼编码表的输出！\n");
}

主函数

int main()
{
	HuffmanTree HT;  //哈夫曼树HT
	HuffmanCode HC;  //哈夫曼编码表HC
	int n, i;         //n是哈夫曼树叶子结点数
	unsigned int* w; //w存放叶子结点权值   
	char j = 'y';
	//程序解说
	
	while (j != 'N' && j != 'n')
	{
		printf("\n\t构造哈夫曼树.\n");
		printf("\n\t首先输入叶子结点数目.\n");
		printf("\n\t然后输入每个叶子结点的权值.\n");
		printf("\n\t程序会构造一棵哈夫曼树并显示哈夫曼编码.\n");
		printf("  5---0110\n 29---10\n  7---1110\n  8---1111\n 14---110\n");
		printf(" 23---00\n  3---0111\n 11---010\n");
		printf("\n请输入叶子结点数目:\n");
		scanf("%d", &n);   //输入叶子结点数
		if (n <= 1) 
		{ 
			printf("\n该数不合理!\n");
			exit(1); 
		}
		w = (unsigned int*)malloc(n * sizeof(unsigned int)); //开辟空间存放权值
		printf("\n请输入各叶子结点的权值:\n");
		for (i = 0;i < n;i++) 
			scanf("%d", &w[i]);       //输入各叶子结点权值
		printf("\n权值输入结束!\n");
		CreatHaffmantree(HT, w, n);             //生成哈夫曼树
		huffmancoding(HT, HC, n);              //进行哈夫曼编码
		printhuffmantree(HC, w, n);             //显示哈夫曼编码
		printf("\n哈夫曼树构造完毕，还要继续吗?(Y/N)\n");
		scanf(" %c", &j);
	}
}

运行结果

哈夫曼树示意图