Question

我试图解决问题not supported

给出一棵二叉树，返回其节点值的级别顺序遍历。（即，从左到右，逐级）。

例如：   给定二叉树[3,9,20,null,null,15,7]，
    3
   / \
  9  20
    /  \
   15   7
将其级别顺序遍历返回为：
[
  [3],
  [9,20],
  [15,7]
]

我的解决方案很直观，BFS遍历并收集每个级别中的值

from collections import deque
class Solution:
    def levelOrder(self, root):
        if not root: return [] #base case 
        res = []
        #queue to colloct all the nodes
        queue = deque([root]) 

        while queue:
            level_vals = [] #hold the values at the current level.
            for _ in range(len(queue)): #evalute once before execution enter the loop
                cur = queue.popleft()
                if node.left:
                    queue.append(cur.left)
                if node.right:
                    queue.append(cur.right)
                level_vals.append(cur.val)
            res.append(level_vals)
        return res

我在“讨论区”中读到了这样的bfs和队列解决方案

# BFS + deque
def levelOrder(self, root):
    res, queue = [], deque([(root, 0)])
    while queue:
        cur, level = queue.popleft()
        if cur:
            if len(res) < level+1:
                res.append([])
            res[level].append(cur.val)
            queue.append((cur.left, level+1))
            queue.append((cur.right, level+1))
    return res

我对条件检查if len(res) < level+1: res.append([])感到困惑，并认为它可以是'

        if cur:
            #if len(res) < level+1:
            res.append([])
            res[level].append(cur.val)
            queue.append((cur.left, level+1))
            queue.append((cur.right, level+1))
    return res

为什么需要条件检查？

Answer 1

可以进行条件检查，以便仅在res中遇到新级别时，才将新数组（对应于新级别）附加到queue。如果没有该检查，则代码将为res中遇到的每个节点向queue追加一个新的空数组。

让我们看看使用条件检查运行代码时会发生什么。对于示例树，首先从队列中弹出值为3的根节点。此时res的长度为0，level的长度也为0。因此len(res) > level + 1是正确的。因此，将一个新的空数组附加到res上，以存储树级别0的值。处理级别1的第一个节点（值为9）时的情况相同。但是，在处理完之后，当我们开始处理级别1的第二个节点（值为20）时，res数组具有2个元素（每个级别一个），级别的值为1。{{1 }}为假，len(res) > level + 1中未插入任何内容。

如果不进行检查，则在迭代结束时res数组将是这样的：

res

请注意，由于树中有5个节点，因此总共将5个数组附加到[ [3], [9, 20], [15, 7], [], [] ]上，但由于树有3个级别，因此仅占据了前3个。

BFS以每个顺序检索值的元素

1 个答案: