构造哈夫曼树,并根据哈夫曼树构造哈夫曼编码
时间: 2023-12-22 09:04:51 浏览: 156
以下是构造哈夫曼树并根据哈夫曼树构造哈夫曼编码的步骤和示例代码:
1. 统计每个字符出现的频率,并将它们存储在一个字典中。
2. 将每个字符及其频率作为一个节点,构造一个森林。
3. 从森林中选出两个频率最小的节点,将它们合并成一个新节点,并将新节点的频率设置为两个节点的频率之和。将新节点插入森林中。
4. 重复步骤3,直到森林中只剩下一个节点,这个节点就是哈夫曼树的根节点。
5. 根据哈夫曼树构造哈夫曼编码。从根节点开始,遍历哈夫曼树,当遇到左子树时,在编码的末尾添加0,当遇到右子树时,在编码的末尾添加1。当遍历到叶子节点时,就得到了该字符的哈夫曼编码。
以下是示例代码:
```python
# 定义节点类
class Node:
def __init__(self, freq, char=None):
self.freq = freq
self.char = char
self.left = None
self.right = None
# 统计字符频率
def count_freq(text):
freq = {}
for char in text:
if char in freq:
freq[char] += 1
else:
freq[char] = 1
return freq
# 构造哈夫曼树
def build_huffman_tree(freq):
forest = [Node(freq[char], char) for char in freq]
while len(forest) > 1:
forest.sort(key=lambda x: x.freq)
node1 = forest.pop(0)
node2 = forest.pop(0)
new_node = Node(node1.freq + node2.freq)
new_node.left = node1
new_node.right = node2
forest.append(new_node)
return forest[0]
# 构造哈夫曼编码
def build_huffman_code(node, code=''):
if node.char:
print(node.char, code)
else:
build_huffman_code(node.left, code + '0')
build_huffman_code(node.right, code + '1')
# 示例
text = 'hello world'
freq = count_freq(text)
tree = build_huffman_tree(freq)
build_huffman_code(tree)
```
输出结果为:
```
d 00
h 01
r 100
e 1010
w 1011
o 110
l 1110
1111
```
阅读全文
相关推荐















