一、Huffman算法介绍
霍夫曼编码(英语:Huffman Coding),又译为哈夫曼编码、赫夫曼编码,是一种用于无损数据压缩的熵编码(权编码)算法。在计算机数据处理中,霍夫曼编码使用变长编码表对源符号(如文件中的一个字母)进行编码,其中变长编码表是通过一种评估来源符号出现几率的方法得到的,出现几率高的字母使用较短的编码,反之出现几率低的则使用较长的编码,这便使编码之后的字符串的平均长度、期望值降低,从而达到无损压缩数据的目的。
霍夫曼树又称最优二叉树,是一种带权路径长度最短的二叉树。所谓树的带权路径长度,就是树中所有的叶结点的权值乘上其到根结点的路径长度(若根结点为0层,叶结点到根结点的路径长度为叶结点的层数)。
前缀代码表示以一种方式分配代码(位序列),以使分配给一个字符的代码不是分配给任何其他字符的代码的前缀。这就是霍夫曼编码如何确保在解码生成的比特流时没有歧义的地方。
让我们通过一个反例来了解前缀代码。假设有四个字符a,b,c和d,它们对应的可变长度代码分别为00、01、0和1。由于分配给c的代码是分配给a和b的代码的前缀,因此这种编码会产生歧义。如果压缩的比特流是0001,则解压缩的输出可以是“ cccd”或“ ccb”或“ acd”或“ ab”。
霍夫曼编码主要包括两个主要部分:
1)根据输入字符构建霍夫曼树。
2)遍历霍夫曼树并将代码分配给字符。
二、构建霍夫曼树的步骤
输入是唯一字符及其出现频率的数组,输出是霍夫曼树。
1. 为每个唯一字符创建一个叶节点,并为所有叶节点建立一个最小堆(Min Heap用作优先级队列。frequency字段的值用于比较最小堆中的两个节点。最初,把最不频繁的字符作为根)
2. 从最小堆中提取频率最小的两个节点。
3. 创建一个频率等于两个节点频率之和的新内部节点。使第一个提取的节点为其左子节点,另一个提取的节点为其右子节点。将此节点添加到最小堆中。
4. 重复步骤2和3,直到堆仅包含一个节点。其余节点是根节点,树已完成。
一个例子:
| 字符 | A | B | C | D | E |
| 频率 | 1 | 7 | 6 | 5 | 2 |
步骤1:构建一个包含5个节点的最小堆,其中每个节点代表具有单个节点的树的根。
步骤2:从最小堆中提取两个最小频率节点。添加一个频率为1 + 2 = 3的新内部节点。

现在最小堆包含4个节点,其中3个节点是每个具有单个元素的结点,一个堆节点是具有3个元素。
| 字符 | 频率 |
| 内部节点 | 3 |
| D | 5 |
| C | 6 |
| B | 7 |
步骤3:从堆中提取两个最低频率节点。添加频率为12 + 13 = 25的新内部节点

现在最小堆包含3个节点,其中2个节点是每个具有单个元素的节点,两个堆节点是具有多个节点的子树。
| 字符 | 频率 |
| 内部节点 | 8 |
| C | 6 |
| B | 7 |
步骤4:提取两个最低频率节点。添加频率为6 + 7 = 13的新内部节点

现在,最小堆包含2个节点。
| 字符 | 频率 |
| 内部节点 | 8 |
| 内部节点 | 13 |
步骤6:提取两个最低频率节点。添加频率为8 + 13 = 21的新内部节点

现在最小堆中只剩下一个节点,因此算法在此处停止。
从霍夫曼树打印代码的步骤:
遍历从根开始形成的树。维护一个辅助阵列。当移到左孩子时,将0写入数组。当移动到正确的孩子时,将1写入数组。遇到叶节点时打印阵列。

输出如下:

源代码:

1 package tree;
2
3 import java.util.Comparator;
4 import java.util.PriorityQueue;
5
6 /**
7 * 这是一个哈夫曼树 (Huffman Tree),用于无损数据压缩的熵编码算法
8 * 哈夫曼树是带权路径长度最短的树,权值较大的结点离根较近。
9 */
10 public class HuffmanTree {
11
12 /**
13 * 通过树遍历的霍夫曼代码
14 * @param root
15 * @param s
16 */
17 public static void printCode(HuffmanNode root, String s) {
18 /* 如果左右为空,那么这就是一个叶子结点 */
19 if (root.left == null && root.right == null && Character.isLetter(root.c)) {
20 System.out.println(root.c + ":" + s);
21 return;
22 }
23 assert root.left != null;
24 printCode(root.left, s + "0");
25 printCode(root.right, s + "1");
26 }
27
28 public static void main(String[] args) {
29
30 int n = 5; // 字符数量
31
32 /* 下面那个例子的最后结果:
33 A:000
34 E:001
35 D:01
36 C:10
37 B:11
38 */
39 char[] charArray = {'A', 'B', 'C', 'D', 'E'};
40 int[] charfreq = {1, 7, 6, 5, 2};
41
42 /* 创建优先级队列q,创建一个最低优先级队列(min-heap)。*/
43 PriorityQueue<HuffmanNode> q = new PriorityQueue<>(n, new MyComparator());
44
45 /* 为每个字符创建一个Huffman树的叶子结点,并 */
46 for (int i = 0; i < n; i++) {
47 /* 创建一个Huffman节点对象,并将其添加到优先级队列。*/
48 HuffmanNode hn = new HuffmanNode();
49
50 hn.c = charArray[i];
51 hn.data = charfreq[i];
52
53 hn.left = null;
54 hn.right = null;
55
56 q.add(hn);
57 }
58
59 /* 创建一个Huffman根节点 */
60 HuffmanNode root = null;
61
62 while (q.size() > 1) {
63 /* 提取第一个最小的。*/
64 HuffmanNode x = q.peek();
65 q.poll();
66
67 /* 提取第二个最小的。*/
68 HuffmanNode y = q.peek();
69 q.poll();
70
71 /* 新建一个新结点,将两个节点的频率之和分配给f节点 */
72 HuffmanNode f = new HuffmanNode();
73 f.data = x.data + y.data;
74 f.c = '-';
75
76 f.left = x;
77 f.right = y;
78
79 root = f;
80
81 q.add(f);
82 }
83
84 printCode(root, "");
85 }
86 }
87
88 /**
89 * 哈夫曼节点
90 */
91 class HuffmanNode {
92 int data;
93 char c;
94
95 HuffmanNode left;
96 HuffmanNode right;
97 }
98
99 /**
100 * 用于比较哈夫曼节点值的大小
101 */
102 class MyComparator implements Comparator<HuffmanNode> {
103
104 @Override
105 public int compare(HuffmanNode o1, HuffmanNode o2) {
106 return o1.data - o2.data;
107 }
108 }
