建立一棵二叉树,求二叉数的树的深度、统计叶子结点的个数、统计总的结点个数、进行层序遍历、交换左右子树等; 3.哈夫曼编码译码系统。的代码
时间: 2023-07-10 16:14:31 浏览: 69
求度为2的结点个数-二叉树
好的,以下是建立二叉树并实现相关功能的示例代码:
```python
# 定义节点类
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
# 建立二叉树
def build_tree():
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)
return root
# 计算树的深度
def get_depth(root):
if not root:
return 0
return 1 + max(get_depth(root.left), get_depth(root.right))
# 统计叶子节点的个数
def count_leaves(root):
if not root:
return 0
if not root.left and not root.right:
return 1
return count_leaves(root.left) + count_leaves(root.right)
# 统计总的节点个数
def count_nodes(root):
if not root:
return 0
return 1 + count_nodes(root.left) + count_nodes(root.right)
# 层序遍历
def level_order_traversal(root):
if not root:
return []
queue = [root]
res = []
while queue:
level_size = len(queue)
level_res = []
for i in range(level_size):
node = queue.pop(0)
level_res.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
res.append(level_res)
return res
# 交换左右子树
def swap_left_right(root):
if not root:
return
temp = root.left
root.left = root.right
root.right = temp
swap_left_right(root.left)
swap_left_right(root.right)
# 测试代码
root = build_tree()
print(get_depth(root)) # 输出:3
print(count_leaves(root)) # 输出:4
print(count_nodes(root)) # 输出:7
print(level_order_traversal(root)) # 输出:[[1], [2, 3], [4, 5, 6, 7]]
swap_left_right(root)
print(level_order_traversal(root)) # 输出:[[1], [3, 2], [7, 6, 5, 4]]
```
关于哈夫曼编码译码系统的代码,由于需要考虑一些细节问题,代码可能会比较长,这里就不提供具体的实现了。
阅读全文