面试题32 II 从上到下打印二叉树 II

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

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

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：

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

 

思路

• 这道题与上一道题很像，只是要求分层打印。因此思路如下：
• 首先复现上一道题的解答，即按BFS遍历并打印；
• 再改为层次遍历输出即可。

 

上一道全部打印题目的代码为：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        res = []
        queue = []

        queue.append(root)
        while queue:
            node = queue[0]
            queue = queue[1:]
            res.append(node.val)
            if queue.left:
                queue.append(queue.left)
            if queue.right:
                queue.append(queue.right)
            
        return res

 

接下来就是如何判断分层的位置，并进行分层输出即可。

其实通过观察可以发现，每次循环都必定会处理一个节点，即将一个节点的值加入res中。因此，只需要在开始处理下一层的时候求一下queue的长度length，也就是queue此时包含的节点数，也就是这一层的节点数，然后连续执行length次循环即可。

 

代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        res = []
        queue = []

        queue.append(root)
        while queue:
            temRes = []
            for i in range(len(queue)):
                node = queue[0]
                queue = queue[1:]
                temRes.append(node.val)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            res.append(temRes)
        return res

 

结果