数据结构之哈夫曼树(C语言)

一、哈夫曼树的概念

给定N个权值作为N个叶子结点,构造一棵二叉树,若该树的带权路径长度达到最小,称这样的二叉树为最优二叉树,也称为哈夫曼树(Huffman Tree)。哈夫曼树是带权路径长度最短的树,权值较大的结点离根较近。

该代码的核心在于弄懂如何构建哈夫曼树和生成哈夫曼树编码。主要通过找出两个最小值开始作为最底端叶节点,由下到上构建出哈夫曼树。

二、代码步骤

1、定义二叉树和字符指针数组

2、二叉树的初始化

3、查找最小两个值的下标(min1<min2)

4、构建哈夫曼树

5、生成哈夫曼树编码

6、测试代码

7、程序入口

8、运行结果


三、代码功能

1、定义二叉树和字符指针数组

typedef struct HTNode
{
	double weight;
	int parent;
	int lc, rc; 
}*HuffmanTree;

typedef char **HuffmanCode;

2、二叉树的初始化

HuffmanTree initHuffmanTree(HuffmanTree& HT,int n)
{
	int i;
	int m = 2 * n - 1;
	HT = (HuffmanTree)malloc(sizeof(HTNode)*(m + 1));
	for(i = 0; i <= m; i++)
	{
		HT[i].lc = 0;
		HT[i].parent = 0;
		HT[i].rc = 0;
		HT[i].weight = 0;
	}
	return HT;
}

3、查找最小两个值的下标(min1<min2)

void Select(HuffmanTree& HT, int n, int& min1, int& min2)
{
	int min;
	for (int i = 1; i <= n; i++)
	{
		if (HT[i].parent == 0)
		{
			min = i;
			break;
		}
	}
	for (int i = min + 1; i <= n; i++)
	{
		if (HT[i].parent == 0 && HT[i].weight < HT[min].weight)
			min = i;
	}
	min1 = min;
	for (int i = 1; i <= n; i++)
	{
		if (HT[i].parent == 0 && i != min1)
		{
			min = i;
			break;
		}
	}
	for (int i = min + 1; i <= n; i++)
	{
		if (HT[i].parent == 0 && HT[i].weight < HT[min].weight&&i != min1)
			min = i;
	}
	min2 = min;
}

4、构建哈夫曼树

void CreateHuff(HuffmanTree& HT,double* w, int n)
{
	int m = 2 * n - 1; 
	for (int i = 1; i <= n; i++)
	{
		HT[i].weight = w[i - 1];
	}
	for (int i = n + 1; i <= m; i++)
	{
		int min1, min2;
		Select(HT, i - 1, min1, min2);
		HT[i].weight = HT[min1].weight + HT[min2].weight; 
		HT[min1].parent = i; 
		HT[min2].parent = i;
		HT[i].lc = min1; 
		HT[i].rc = min2;
	}
	printf("Huffman is: \n");
	printf("subscript    weight     parent   lchild   rchild\n");
	printf("0                                  \n");
	for (int i = 1; i <= m; i++)
	{
	    printf("%-4d         %-6.2lf      %-6d     %-6d   %-6d\n", i, HT[i].weight, HT[i].parent, HT[i].lc, HT[i].rc);
	}
	printf("\n");
}

5、生成哈夫曼树编码

void HuffCoding(HuffmanTree& HT, HuffmanCode& HC, int n)
{
	HC = (HuffmanCode)malloc(sizeof(char*)*(n + 1));
	char* code = (char*)malloc(sizeof(char)*n);
	code[n - 1] = '\0';
	for (int i = 1; i <= n; i++)
	{
		int start = n - 1; 
		int c = i; 
		int p = HT[c].parent; 
		while (p)
		{
			if (HT[p].lc == c) 
				code[--start] = '0';
			else
				code[--start] = '1';
			c = p;
			p = HT[c].parent; 
		}
		HC[i] = (char*)malloc(sizeof(char)*(n - start));
		strcpy(HC[i], &code[start]); 
	}
	free(code);
}

6、测试代码

void test()
{
	int n = 0;
	printf("input the number of data: ");
	scanf("%d", &n);
	double* w = (double*)malloc(sizeof(double)*n);
	if (n >= 0)
	{
		printf("input the data: ");
	    for (int i = 0; i < n; i++)
	    {
	    	scanf("%lf", &w[i]);
	    }
    	HuffmanTree HT;
    	HT = initHuffmanTree(HT,n);
    	CreateHuff(HT, w, n); 

    	HuffmanCode HC;
    	HuffCoding(HT, HC, n); 

    	for (int i = 1; i <= n; i++)
    	{
    		printf("The data %.2lf code is:  %s\n", HT[i].weight, HC[i]);
    	}
    	free(w);	
	}
	else
	{
		printf("malloc fail\n");
		return;
	}
}

7、程序入口

int main()
{
	test();
	return 1;
}

8、运行结果

input the number of data: 5
input the data: 0.1 0.2 0.3 0.25 0.15
Huffman is:
subscript    weight     parent   lchild   rchild
0
1            0.10        6          0        0
2            0.20        7          0        0
3            0.30        8          0        0
4            0.25        7          0        0
5            0.15        6          0        0
6            0.25        8          1        5
7            0.45        9          2        4
8            0.55        9          6        3
9            1.00        0          7        8

The data 0.10 code is:  100
The data 0.20 code is:  00
The data 0.30 code is:  11
The data 0.25 code is:  01
The data 0.15 code is:  101

四、整体代码

#include <stdio.h>
#include <malloc.h>
#include <string.h>

typedef struct HTNode
{
	double weight;
	int parent;
	int lc, rc; 
}*HuffmanTree;

typedef char **HuffmanCode;

HuffmanTree initHuffmanTree(HuffmanTree& HT,int n)
{
	int i;
	int m = 2 * n - 1;
	HT = (HuffmanTree)malloc(sizeof(HTNode)*(m + 1));
	for(i = 0; i <= m; i++)
	{
		HT[i].lc = 0;
		HT[i].parent = 0;
		HT[i].rc = 0;
		HT[i].weight = 0;
	}
	return HT;
}

void Select(HuffmanTree& HT, int n, int& min1, int& min2)
{
	int min;
	for (int i = 1; i <= n; i++)
	{
		if (HT[i].parent == 0)
		{
			min = i;
			break;
		}
	}
	for (int i = min + 1; i <= n; i++)
	{
		if (HT[i].parent == 0 && HT[i].weight < HT[min].weight)
			min = i;
	}
	min1 = min;
	for (int i = 1; i <= n; i++)
	{
		if (HT[i].parent == 0 && i != min1)
		{
			min = i;
			break;
		}
	}
	for (int i = min + 1; i <= n; i++)
	{
		if (HT[i].parent == 0 && HT[i].weight < HT[min].weight&&i != min1)
			min = i;
	}
	min2 = min;
}

void CreateHuff(HuffmanTree& HT,double* w, int n)
{
	int m = 2 * n - 1; 
	for (int i = 1; i <= n; i++)
	{
		HT[i].weight = w[i - 1];
	}
	for (int i = n + 1; i <= m; i++)
	{
		int min1, min2;
		Select(HT, i - 1, min1, min2);
		HT[i].weight = HT[min1].weight + HT[min2].weight; 
		HT[min1].parent = i; 
		HT[min2].parent = i;
		HT[i].lc = min1; 
		HT[i].rc = min2;
	}
	printf("Huffman is: \n");
	printf("subscript    weight     parent   lchild   rchild\n");
	printf("0                                  \n");
	for (int i = 1; i <= m; i++)
	{
	    printf("%-4d         %-6.2lf      %-6d     %-6d   %-6d\n", i, HT[i].weight, HT[i].parent, HT[i].lc, HT[i].rc);
	}
	printf("\n");
}

void HuffCoding(HuffmanTree& HT, HuffmanCode& HC, int n)
{
	HC = (HuffmanCode)malloc(sizeof(char*)*(n + 1));
	char* code = (char*)malloc(sizeof(char)*n);
	code[n - 1] = '\0';
	for (int i = 1; i <= n; i++)
	{
		int start = n - 1; 
		int c = i; 
		int p = HT[c].parent; 
		while (p)
		{
			if (HT[p].lc == c) 
				code[--start] = '0';
			else
				code[--start] = '1';
			c = p;
			p = HT[c].parent; 
		}
		HC[i] = (char*)malloc(sizeof(char)*(n - start));
		strcpy(HC[i], &code[start]); 
	}
	free(code);
}

void test()
{
	int n = 0;
	printf("input the number of data: ");
	scanf("%d", &n);
	double* w = (double*)malloc(sizeof(double)*n);
	if (n >= 0)
	{
		printf("input the data: ");
	    for (int i = 0; i < n; i++)
	    {
	    	scanf("%lf", &w[i]);
	    }
    	HuffmanTree HT;
    	HT = initHuffmanTree(HT,n);
    	CreateHuff(HT, w, n); 

    	HuffmanCode HC;
    	HuffCoding(HT, HC, n); 

    	for (int i = 1; i <= n; i++)
    	{
    		printf("The data %.2lf code is:  %s\n", HT[i].weight, HC[i]);
    	}
    	free(w);	
	}
	else
	{
		printf("malloc fail\n");
		return;
	}
}

int main()
{
	test();
	return 1;
}

  • 10
    点赞
  • 129
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
以下是哈夫曼树C语言实现代码: ```c #include <stdio.h> #include <stdlib.h> #include <string.h> #define MAX_TREE_HT 100 struct MinHeapNode { char data; unsigned freq; struct MinHeapNode *left, *right; }; struct MinHeap { unsigned size; unsigned capacity; struct MinHeapNode **array; }; struct MinHeapNode* newNode(char data, unsigned freq) { struct MinHeapNode* node = (struct MinHeapNode*) malloc(sizeof(struct MinHeapNode)); node->left = node->right = NULL; node->data = data; node->freq = freq; return node; } struct MinHeap* createMinHeap(unsigned capacity) { struct MinHeap* minHeap = (struct MinHeap*) malloc(sizeof(struct MinHeap)); minHeap->size = 0; minHeap->capacity = capacity; minHeap->array = (struct MinHeapNode**) malloc(minHeap->capacity * sizeof(struct MinHeapNode*)); return minHeap; } void swapMinHeapNode(struct MinHeapNode** a, struct MinHeapNode** b) { struct MinHeapNode* t = *a; *a = *b; *b = t; } void minHeapify(struct MinHeap* minHeap, int idx) { int smallest = idx; int left = 2 * idx + 1; int right = 2 * idx + 2; if (left < minHeap->size && minHeap->array[left]->freq < minHeap->array[smallest]->freq) smallest = left; if (right < minHeap->size && minHeap->array[right]->freq < minHeap->array[smallest]->freq) smallest = right; if (smallest != idx) { swapMinHeapNode(&minHeap->array[smallest], &minHeap->array[idx]); minHeapify(minHeap, smallest); } } int isSizeOne(struct MinHeap* minHeap) { return (minHeap->size == 1); } struct MinHeapNode* extractMin(struct MinHeap* minHeap) { struct MinHeapNode* temp = minHeap->array[0]; minHeap->array[0] = minHeap->array[minHeap->size - 1]; --minHeap->size; minHeapify(minHeap, 0); return temp; } void insertMinHeap(struct MinHeap* minHeap, struct MinHeapNode* minHeapNode) { ++minHeap->size; int i = minHeap->size - 1; while (i && minHeapNode->freq < minHeap->array[(i - 1) / 2]->freq) { minHeap->array[i] = minHeap->array[(i - 1) / 2]; i = (i - 1) / 2; } minHeap->array[i] = minHeapNode; } void buildMinHeap(struct MinHeap* minHeap) { int n = minHeap->size - 1; int i; for (i = (n - 1) / 2; i >= 0; --i) minHeapify(minHeap, i); } void printArr(int arr[], int n) { int i; for (i = 0; i < n; ++i) printf("%d", arr[i]); printf("\n"); } int isLeaf(struct MinHeapNode* root) { return !(root->left) && !(root->right); } struct MinHeap* createAndBuildMinHeap(char data[], int freq[], int size) { struct MinHeap* minHeap = createMinHeap(size); for (int i = 0; i < size; ++i) minHeap->array[i] = newNode(data[i], freq[i]); minHeap->size = size; buildMinHeap(minHeap); return minHeap; } struct MinHeapNode* buildHuffmanTree(char data[], int freq[], int size) { struct MinHeapNode *left, *right, *top; struct MinHeap* minHeap = createAndBuildMinHeap(data, freq, size); while (!isSizeOne(minHeap)) { left = extractMin(minHeap); right = extractMin(minHeap); top = newNode('$', left->freq + right->freq); top->left = left; top->right = right; insertMinHeap(minHeap, top); } return extractMin(minHeap); } void printCodes(struct MinHeapNode* root, int arr[], int top) { if (root->left) { arr[top] = 0; printCodes(root->left, arr, top + 1); } if (root->right) { arr[top] = 1; printCodes(root->right, arr, top + 1); } if (isLeaf(root)) { printf("%c: ", root->data); printArr(arr, top); } } void HuffmanCodes(char data[], int freq[], int size) { struct MinHeapNode* root = buildHuffmanTree(data, freq, size); int arr[MAX_TREE_HT], top = 0; printCodes(root, arr, top); } int main() { char arr[] = {'a', 'b', 'c', 'd', 'e', 'f'}; int freq[] = {5, 9, 12, 13, 16, 45}; int size = sizeof(arr) / sizeof(arr[0]); HuffmanCodes(arr, freq, size); return 0; } ``` 该代码实现了哈夫曼树的构建和编码过程。其中,创建和构建最小堆的代码已经实现了,因此主要实现了哈夫曼树的构建和编码的函数。在HuffmanCodes函数中调用buildHuffmanTree函数构建哈夫曼树,然后调用printCodes函数打印每个字符的编码。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值