用Python求二叉树的层次遍历

二叉树的层次遍历是一种广度优先搜索，可以使用队列来实现。我们首先将根节点入队，然后依次取出队列中的节点，将其左右子节点入队，直到队列为空。

下面是一个求解二叉树层次遍历的示例代码：

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

def level_order(root):
    if root is None:
        return []
    res = []
    queue = [root]
    while queue:
        level = []
        for i in range(len(queue)):
            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

# 示例
#       1
#      / \
#     2   3
#    / \
#   4   5

root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)

print(level_order(root))  # 输出 [[1], [2, 3], [4, 5]]

在上面的示例中，我们定义了一个 `level_order` 函数，接受一个二叉树的根节点作为参数，返回二叉树的层次遍历结果。在函数内部，我们首先判断根节点是否为空，如果为空，则返回空列表。如果根节点不为空，我们创建一个空列表 `res` 用于存储层次遍历结果，并创建一个队列 `queue`，将根节点入队。然后，我们循环遍历队列中的节点，依次取出节点并存储其值，将其左右子节点入队，直到队列为空。最后，我们将当前层的节点值列表 `level` 存储到 `res` 中，并继续下一层的遍历，直到遍历完整个二叉树。最后，我们创建了一个二叉树并调用 `level_order` 函数来求解其层次遍历结果，输出结果为 `[[1], [2, 3], [4, 5]]`。