動機

複習bfs

Problem

Given the root of a binary tree, return the level order traversal of its nodes' values. (i.e., from left to right, level by level).

 

Example 1:

Input: root = [3,9,20,null,null,15,7]Output: [[3],[9,20],[15,7]]

Example 2:

Input: root = [1]Output: [[1]]

Example 3:

Input: root = []Output: []

 

Constraints:

  • The number of nodes in the tree is in the range [0, 2000].
  • -1000 <= Node.val <= 1000

Sol

以level為單位,當成state 這樣在解103會有好處

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        q = deque([[root]])
        ret = []
        while q:
            now = []
            rs = q.popleft()
            ret.append([r.val for r in rs])
            for r in rs:
                for n in [r.left,r.right]:
                    if n:
                        now.append(n)
            if len(now) > 0:
                q.append(now)
        return ret