在 Python 中遍历树

树是一种非常常见的数据结构，Python 中也有很多用于表示树的结构，如列表、字典、类等。在 Python 中遍历树通常会用到递归的方法，下面将介绍如何在 Python 中遍历树。

1. 递归遍历二叉树

二叉树是一种特殊的树，每个节点最多只能有两个子节点。因此，在遍历二叉树时，我们可以使用递归方法来处理每个节点。

下面是一个简单的二叉树节点的定义：

class TreeNode:
    def __init__(self, val=None, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

其中，val 是节点的值，left 和 right 分别是该节点的左右子节点。

对于二叉树的遍历有三种方式：前序遍历、中序遍历、后序遍历。具体的遍历思路及代码如下：

前序遍历

前序遍历的遍历顺序是根节点、左子树、右子树。

def preorderTraversal(root):
    if root:
        print(root.val, end=" ")
        preorderTraversal(root.left)
        preorderTraversal(root.right)

中序遍历

中序遍历的遍历顺序是左子树、根节点、右子树。

def inorderTraversal(root):
    if root:
        inorderTraversal(root.left)
        print(root.val, end=" ")
        inorderTraversal(root.right)

后序遍历

后序遍历的遍历顺序是左子树、右子树、根节点。

def postorderTraversal(root):
    if root:
        postorderTraversal(root.left)
        postorderTraversal(root.right)
        print(root.val, end=" ")

2. 递归遍历普通树

对于普通树，我们可以将其看作一棵多叉树，同样可以使用递归方法处理每个节点。下面是一个简单的普通树节点的定义：

class TreeNode:
    def __init__(self, val=None, children=None):
        self.val = val
        self.children = children if children else []

其中，val 是节点的值，children 是该节点的子节点列表。

对于普通树的遍历，我们可以使用类似于遍历二叉树的方法来处理，只需将递归遍历左右子树的方式修改为遍历所有子节点即可。具体的遍历思路及代码如下：

前序遍历

前序遍历的遍历顺序是根节点、子节点1、子节点2、……、子节点n。

def preorderTraversal(root):
    if root:
        print(root.val, end=" ")
        for child in root.children:
            preorderTraversal(child)

后序遍历

后序遍历的遍历顺序是子节点1、子节点2、……、子节点n、根节点。

def postorderTraversal(root):
    if root:
        for child in root.children:
            postorderTraversal(child)
        print(root.val, end=" ")

总结

以上介绍了在 Python 中遍历树的方法，其中，递归方法是最常用的方式。在遍历树时，要注意对节点为空的情况进行处理，以避免产生异常。