二叉树的层序遍历的两种方法，递归和迭代，附代码实现

前端

2024-01-15 14:42:47

层序遍历：广度优先遍历二叉树

在处理二叉树数据结构时，层序遍历是一种至关重要的遍历方法。它不同于深度优先遍历（例如前序遍历、中序遍历和后序遍历），而是采用广度优先策略，从根节点开始逐层遍历树。

层序遍历的递归解法

想象一下一棵二叉树被分割成多个水平层，每一层包含同一层级的节点。层序遍历的递归解法遵循一个简单而有效的步骤：

将根节点添加到队列中。
循环处理队列中的节点：
- 从队列中弹出最前面的节点。
- 将该节点的左右孩子（如果有的话）添加到队列中。
- 访问该节点（打印或进行其他操作）。

层序遍历的迭代解法

与递归解法类似，迭代解法不需要使用递归函数。它采用以下步骤：

将根节点添加到队列中。
循环处理队列中的节点：
- 从队列中弹出最前面的节点。
- 将该节点的左右孩子（如果有的话）添加到队列中。
- 将该节点添加到结果列表中。

代码实现

Python：

递归解法：

def level_order_traversal_recursive(root):
    if not root:
        return

    queue = [root]
    while queue:
        node = queue.pop(0)
        print(node.val)
        if node.left:
            queue.append(node.left)
        if node.right:
            queue.append(node.right)

迭代解法：

def level_order_traversal_iterative(root):
    if not root:
        return

    queue = [root]
    result = []
    while queue:
        node = queue.pop(0)
        result.append(node.val)
        if node.left:
            queue.append(node.left)
        if node.right:
            queue.append(node.right)

    return result