107 - 二叉树的层次遍历 II - python

36 阅读1分钟

给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

例如:给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其自底向上的层次遍历为:

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

这道题和一般的层序遍历是相同的,只不过在最后返回结果之前做一次数组的反转,使其满足自底向上的层次遍历的条件。

AC code:

class Solution:
    def levelOrderBottom(self, root: TreeNode) -> List[List[int]]:
        if root == None: return []

        queue = [root]
        res = []
        while queue:
            cur = []
            nextLayer = []
            for node in queue:
                cur.append(node.val)
                if node.left:
                    nextLayer.append(node.left)
                if node.right:
                    nextLayer.append(node.right)

            res.append(cur)
            queue = nextLayer

        res.reverse()
        return res