层序遍历python

要实现一个层序遍历的代码,并获取每一层的节点数据,你可以使用广度优先搜索(BFS)的方法。下面是一个 Python 示例,该代码通过队列完成层序遍历,将结果逐层收集到一个列表中。

这里的代码将打印每一层的节点,而不是仅仅是右侧视图:

from collections import deque
from typing import List, Optional

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

def levelOrder(root: Optional[TreeNode]) -> List[List[int]]:
    if not root:
        return []

    result = []
    queue = deque([root])

    while queue:
        level_size = len
### Python 实现二叉树层序遍历 #### 方法概述 层序遍历是一种基于广度优先搜索(BFS)的二叉树遍历方式。它按照从上到下、从左到右的顺序访问每一层节点。为了实现这一目标,通常会借助队列来存储待处理的节点。 #### 代码实现 以下是一个完整的 Python 实现: ```python from collections import deque class TreeNode: def __init__(self, value): self.val = value self.left = None self.right = None def levelOrder(root): if not root: # 如果根节点为空,则直接返回空列表 return [] result = [] # 存储最终的结果 queue = deque([root]) # 初始化队列并加入根节点 while queue: # 当队列不为空时继续循环 current_level_size = len(queue) # 获取当前层的节点数量 current_level_values = [] # 存储当前层的值 for _ in range(current_level_size): # 遍历当前层的所有节点 node = queue.popleft() # 取出队首节点 current_level_values.append(node.val) # 将其值加入当前层结果 if node.left: # 如果存在左子节点,将其加入队列 queue.append(node.left) if node.right: # 如果存在右子节点,将其加入队列 queue.append(node.right) result.append(current_level_values) # 将当前层的结果加入总结果中 return result ``` 此代码通过 `deque` 数据结构实现了高效的队列操作,并能够正确处理满二叉树和非满二叉树的情况[^1]。 #### 示例运行 假设我们构建了一棵如下所示的二叉树: ``` 1 / \ 2 3 / \ \ 4 5 6 / 7 ``` 对应的层序遍历结果应为:[[1], [2, 3], [4, 5, 6], [7]]。 可以验证上述代码能正确输出预期结果。 --- #### 易错点分析 在实现过程中需要注意以下几个常见错误: 1. **未初始化队列**:如果忘记将根节点放入队列,程序可能陷入死循环或提前结束。 2. **忽略空节点**:某些情况下需要显式表示空节点的位置,具体取决于需求[^4]。 3. **层次划分不清**:如果没有记录每层的节点数,在嵌套结构复杂的树中可能会混淆不同层的数据[^3]。 以上问题都可以通过仔细设计逻辑加以规避。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值