算法笔记练习 9.8 哈夫曼树 问题 B: 算法6-13：自顶向下的赫夫曼编码

算法笔记练习 题解合集

本题链接

题目

题目描述
在本题中，我们将要讨论的是自顶向下的赫夫曼编码算法。从根出发，遍历整棵赫夫曼树从而求得各个叶子结点所表示的字符串。算法的关键部分可以表示如下：

在本题中，读入n个字符所对应的权值，生成赫夫曼编码，并依次输出计算出的每一个赫夫曼编码。

输入
输入的第一行包含一个正整数n，表示共有n个字符需要编码。其中n不超过100。
第二行中有n个用空格隔开的正整数，分别表示n个字符的权值。

输出
共n行，每行一个字符串，表示对应字符的赫夫曼编码。

样例输入

8
5 29 7 8 14 23 3 11

样例输出

0110
10
1110
1111
110
00
0111
010

思路

思路参考原题中的 C 语言代码即可，如果对于哈夫曼编码的生成过程有疑惑，可以用纸笔画一个简单的哈夫曼树，对着代码过一遍流程应该就懂了。

我把 C 语言代码转成了 C++，注意：

1. 下标的细节，这个题里我的huffmanTree数组和heffmanCode数组的下标都是从 1 开始的。
2. 在生成heffmanCode的时候，第 76 行temp = temp.substr(0, temp.size() - 1);是安全的写法，如果你删去最后一个字符用的是erase，记得要检查字符串是否为空，空字符串用erase会出错。

代码

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>

struct Node {
	int w, parent, lchild, rchild;
};

void findMinTwo(std::vector<Node>& huffman, int& min1, int& min2) {
	int i = 1, w1, w2;
	while (huffman[i++].parent != 0)
		continue;
	min1 = i - 1;
	while (huffman[i++].parent != 0)
		continue;
	min2 = i - 1;
	for ( ; i < huffman.size(); ++i) { 
		if (huffman[i].parent == 0 && huffman[i].w < std::max(huffman[min1].w, huffman[min2].w)) { 
			if (huffman[min1].w != huffman[min2].w)
				(huffman[min1].w > huffman[min2].w ? min1 : min2) = i;
			else
				(min1 > min2 ? min1 : min2) = i; 
		}
	}
	if (min1 > min2)
		std::swap(min1, min2);
} 

int main() {
	int n, input, min1, min2;
	std::vector<Node> huffmanTree;
	std::vector<std::string> huffmanCode;
	while (scanf("%d", &n) != EOF) {
		if (n == 1) {
			scanf("%d", &input);
			printf("1\n");
			continue;
		} 
		huffmanTree.clear();
		huffmanTree.resize(n + 1);
		huffmanTree[0] = {0, 0, 0, 0};
		huffmanCode.clear();
		huffmanCode.resize(n + 1);
		for (int i = 1; i <= n; ++i) {
			scanf("%d", &input);
			huffmanTree[i] = {input, 0, 0, 0};
		}
		for (int i = n + 1; i <= 2 * n - 1; ++i) {
			findMinTwo(huffmanTree, min1, min2);
			huffmanTree[min1].parent = huffmanTree[min2].parent = i;
			huffmanTree.push_back( {huffmanTree[min1].w + huffmanTree[min2].w, 0, min1, min2} );
		}
		for (int i = 1; i <= 2 * n - 1; ++i)
			huffmanTree[i].w = 0;
		int c = 2 * n - 1;
		std::string temp;
		while (c) {
			if (huffmanTree[c].w == 0) {
				huffmanTree[c].w = 1;
				if (huffmanTree[c].lchild != 0) {
					c = huffmanTree[c].lchild;
					temp += '0';
				} else if (huffmanTree[c].rchild == 0) {
					huffmanCode[c] = temp; 
				} 
			} else if (huffmanTree[c].w == 1) {
				huffmanTree[c].w = 2;
				if (huffmanTree[c].rchild != 0) {
					c = huffmanTree[c].rchild;
					temp += '1';
				} 
			} else {
				huffmanTree[c].w = 0;
				c = huffmanTree[c].parent;
				temp = temp.substr(0, temp.size() - 1);
			} 
		}
		for (int i = 1; i <= n; ++i)
			std::cout << huffmanCode[i] << '\n';
	} 
	return 0;
}