Python实现遍历树形结构的方法

一、什么是树形结构

树是一种抽象的数据结构，它是由n（n>=1）个有限节点组成一个具有层次关系的集合。其中，一个节点被称作根节点，每个节点至多有一个父节点，但却可以有多个子节点（允许为空）。

树形结构非常适合用来表示具有层次关系的数据，比如文件系统的目录结构，公司组织架构等。在实际应用中，经常需要对树形结构进行遍历、查找等操作。

Python中可以使用类来定义一个节点，并在节点类中定义各种方法，以实现树形结构的操作。下面我们来看看如何实现树的遍历操作。

二、树的遍历方式

在对树形结构进行操作时，遍历是最基本的操作之一。遍历树就是按照一定的顺序，依次访问每个节点及其包含的所有子节点。常用的树的遍历方式有以下三种：

先序遍历（前序遍历）：先遍历根节点，再递归遍历左子树和右子树。
中序遍历：先递归遍历左子树，再遍历根节点，最后递归遍历右子树。
后序遍历：先递归遍历左子树，再递归遍历右子树，最后遍历根节点。

通过深度优先搜索算法，可以实现树的遍历。接下来我们将给出Python代码，实现深度优先搜索算法，实现三种遍历方式的操作。

三、先序、中序、后序遍历的实现


# 定义一个节点类
class Node:
    def __init__(self, val=None, left=None, right=None):
        self.val = val  # 节点的值
        self.left = left  # 左子树
        self.right = right  # 右子树

# 先序遍历
def preorder(root):
    if not root:
        return []
    res = []
    res.append(root.val)
    res += preorder(root.left)
    res += preorder(root.right)
    return res

# 中序遍历
def inorder(root):
    if not root:
        return []
    res = []
    res += inorder(root.left)
    res.append(root.val)
    res += inorder(root.right)
    return res

# 后序遍历
def postorder(root):
    if not root:
        return []
    res = []
    res += postorder(root.left)
    res += postorder(root.right)
    res.append(root.val)
    return res

# 测试代码
root = Node(1, Node(2, Node(4), Node(5)), Node(3, Node(6), Node(7)))
print("先序遍历：", preorder(root))  # [1, 2, 4, 5, 3, 6, 7]
print("中序遍历：", inorder(root))   # [4, 2, 5, 1, 6, 3, 7]
print("后序遍历：", postorder(root)) # [4, 5, 2, 6, 7, 3, 1]

四、层序遍历的实现

层序遍历（又称广度优先遍历）是指在按照树的层次依次遍历树中各节点的过程中，按照先遍历当前层节点，再遍历下一层节点的顺序依次进行。

下面给出Python代码实现：


# 层序遍历
def levelorder(root):
    queue = [root]
    res = []
    if not root:
        return []
    while queue:
        level = []  # 保存当前层的节点值
        size = len(queue)  # 当前层的节点数
        for i in range(size):
            node = queue.pop(0)
            level.append(node.val)
            if node.left:
                queue.append(node.left)
            if node.right:
                queue.append(node.right)
        res.append(level)
    return res

# 测试代码
root = Node(1, Node(2, Node(4), Node(5)), Node(3, Node(6), Node(7)))
print("层序遍历：", levelorder(root)) # [[1], [2, 3], [4, 5, 6, 7]]

五、对树进行查找

树的查找操作，常用的是深度优先搜索和广度优先搜索算法。在深度优先搜索算法中，常用的是先序遍历，因为先序遍历先访问根节点。广度优先搜索中常用的是层序遍历，因为层序遍历是按层遍历节点的，很容易找到需要的节点。

下面给出一个例子，使用深度优先搜索算法，在树中查找是否有某一个节点值为val。


# 深度优先搜索查找树中是否有节点值为val
def dfs(root, val):
    if not root:
        return False
    if root.val == val:
        return True
    return dfs(root.left, val) or dfs(root.right, val)

# 测试代码
root = Node(1, Node(2, Node(4), Node(5)), Node(3, Node(6), Node(7)))
print("查找节点5：", dfs(root, 5)) # True
print("查找节点8：", dfs(root, 8)) # False

六、总结

树形结构在现实中应用非常广泛，比如文件系统、公司架构、数据结构中的优先队列等等。本文主要介绍了树的遍历方式，包括先序遍历、中序遍历、后序遍历和层序遍历，以及深度优先遍历查找操作。通过Python代码的实现，我们可以更好地理解树的遍历、查找等操作。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/291595.html