class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def level_order_traversal(root): if not root: return [] result = [] #存储遍历结果 queue = [root] 使用队列辅助遍历,初始将根节点加入队列 while queue: level_size = len(queue) 当前层的节点个数 level_nodes = [] 存储当前层的节点值 遍历当前层的节点 for _ in range(level_size): 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
点赞
牛客网
牛客企业服务