102. 二叉树的层序遍历

题目

给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:二叉树:[3,9,20,null,null,15,7],

  1. 3
  2. / \
  3. 9 20
  4. / \
  5. 15 7

返回其层序遍历结果:

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

题解

和之前的差不多,都是套广搜的模板

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
from collections import deque
class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        Q=deque()
        L=[]
        if root:
            Q.append(root)
            L.append([root.val])
        while Q:
            L_val=[]
            for _ in range(len(Q)):
                i=Q.popleft()
                if i.left:
                    Q.append(i.left)
                    L_val.append(i.left.val)
                if i.right:
                    Q.append(i.right)
                    L_val.append(i.right.val)
            if L_val:
                L.append(L_val)
        return L