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

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

难度：简单

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

例如:

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

3

/ \

9 20

/ \

15 7

返回其层次遍历结果：

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

# 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]]:
        # BFS
        
        # special condition
        if not root: return []  
        
        res = []
        queue = [root]

        while queue:
            # define "tmp" list to store the node.val of current level
            tmp = []
            queue_len = len(queue)         
            for i in range(queue_len):       
                # pop from queue, and added the value to tmp
                node = queue.pop(0)        
                tmp.append(node.val)
                # add the left and right of node to the queue! if exists
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            # add to res
            res.append(tmp)      

        return res