Question

我正在尝试解决以下问题，比如我有这个二叉树......

然后我需要获得每个级别的所有节点，所以我的结果将是......

[[3],[9,20],[15,7]]

我相信我已接近解决方案，但我不确定我哪里出错了，如果有人可以帮我解决那个很棒的解决方案，如果我走错了路，请让我知道。

我的第一步是使用以下函数获取树的深度......

def get_depth(self, root):
    if root.left == None and root.right == None:
        return 1
    return max(self.get_depth(root.left), self.get_depth(root.right)) + 1

深度为3。

接下来，我调用了一个旨在给我预期结果的函数......

def levelOrder(self, root):
    depth = self.get_depth(root)
    return self.find_comb(root, [[]]*depth, 0)

def find_comb(self, root, result, level):
    if root is None:
        return None
    self.find_comb(root.left, result, level+1)
    self.find_comb(root.right, result, level+1)
    result[level].append(root.val)
    return result

我的思维过程是，我将通过树递归，level参数将跟踪我当前的水平。然后我会将该级别的所有root.val附加到该索引处的结果。

所以，假设我在等级1（我们从0开始），那么1等级的节点是9和20。所以结果看起来像[[], [9, 20], [], []]，它会为树的每个级别执行此操作。我希望我的逻辑清楚。相反，我得到的结果是......

[[9, 15, 7, 20, 3], [9, 15, 7, 20, 3], [9, 15, 7, 20, 3]]

Answer 1

你实际上并不需要找到树的深度。你只是遍历树，同时保持你的水平，让我们在level变量中说。并在listOfLists[level]处插入您的值。当然，你必须处理IndexError: list index out of range。这是一个简单的实现：

def levelTraversal(root, level):
    if root is None:
        return

    if level >= len(listOfLists):
        list = []
        listOfLists.append(list)
    listOfLists[level].append(root.v)
    levelTraversal(root.l, level+1)
    levelTraversal(root.r, level+1)

这样称呼：levelTraversal(rootNode, 0)

对于您的树，结果为：[[3], [9, 20], [15, 7]]

Answer 2

我不知道Phyton，但你可能想看一下Phyton的BFS实现。

https://www.geeksforgeeks.org/breadth-first-traversal-for-a-graph/

二叉树获取每个级别的所有节点

2 个答案: