python实现二叉树的横向遍历

在Python中，实现二叉树的横向遍历，也称为水平遍历或广度优先搜索（Breadth First Search, BFS），通常使用队列数据结构来存储每一层的节点。以下是使用迭代方式实现的一种常见方法：

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

def levelOrder(root):
    if not root:
        return []
    
    queue = [root]  # 队列存放当前层的所有节点
    result = []  # 存放遍历结果

    while queue:  # 当队列非空时，继续遍历
        level_nodes = []  # 保存这一层的节点值
        for _ in range(len(queue)):  # 取出队列里的所有节点，并将它们的左右孩子加入到队列里
            node = queue.pop(0)
            level_nodes.append(node.val)

            if node.left:
                queue.append(node.left)
            if node.right:
                queue.append(node.right)

        result.append(level_nodes)  # 将处理完的这一层的节点值添加到结果列表
    return result

# 示例
# 树节点构建示例
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)

# 水平遍历结果应该是 [[1], [2, 3], [4, 5]]
print(levelOrder(root))