Python二叉树用法介绍

news2024/9/23 11:24:26

更多资料获取

📚 个人网站：ipengtao.com

二叉树是一种常见的数据结构，具有树形结构，每个节点最多有两个子节点。Python中有多种方式来表示和操作二叉树，本文将介绍二叉树的基本概念、构建、遍历和一些常见操作，以及示例代码。

二叉树基本概念

结构

二叉树由节点构成，每个节点可能有左子节点、右子节点和一个父节点。节点之间的关系形成了树形结构，其中根节点是树的顶部。

类型

根据节点数，二叉树可以分为满二叉树、完全二叉树和平衡二叉树等不同类型。这些类型根据节点的特定排列和组织规则进行定义。

Python中二叉树的表示

Python中可以使用类来表示二叉树节点，然后通过节点之间的链接来构建二叉树结构。

以下是一个简单的示例代码，演示了如何创建一个二叉树节点。

class Node:
    def __init__(self, value):
        self.value = value
        self.left = None
        self.right = None

# 创建二叉树节点
root = Node(1)
root.left = Node(2)
root.right = Node(3)
root.left.left = Node(4)
root.left.right = Node(5)

这段代码创建了一个简单的二叉树，其中根节点为1，左子节点为2，右子节点为3，2的左子节点为4，右子节点为5。

二叉树遍历

1. 前序遍历（Preorder Traversal）

前序遍历按照“根左右”的顺序访问二叉树节点。

def preorder_traversal(node):
    if node:
        print(node.value)
        preorder_traversal(node.left)
        preorder_traversal(node.right)

# 执行前序遍历
print("Preorder Traversal:")
preorder_traversal(root)

2. 中序遍历（Inorder Traversal）

中序遍历按照“左根右”的顺序访问二叉树节点。

def inorder_traversal(node):
    if node:
        inorder_traversal(node.left)
        print(node.value)
        inorder_traversal(node.right)

# 执行中序遍历
print("Inorder Traversal:")
inorder_traversal(root)

3. 后序遍历（Postorder Traversal）

后序遍历按照“左右根”的顺序访问二叉树节点。

def postorder_traversal(node):
    if node:
        postorder_traversal(node.left)
        postorder_traversal(node.right)
        print(node.value)

# 执行后序遍历
print("Postorder Traversal:")
postorder_traversal(root)

二叉树常见操作

1. 查找节点

def find_node(node, value):
    if node is None or node.value == value:
        return node
    left = find_node(node.left, value)
    right = find_node(node.right, value)
    return left if left else right

# 查找节点值为3的节点
found_node = find_node(root, 3)
if found_node:
    print(f"Found node with value 3: {found_node.value}")
else:
    print("Node with value 3 not found")

2. 计算树的高度

def tree_height(node):
    if node is None:
        return 0
    else:
        left_height = tree_height(node.left)
        right_height = tree_height(node.right)
        return max(left_height, right_height) + 1

height = tree_height(root)
print(f"Tree height is: {height}")