LeetCode - 102. 二叉树的层序遍历（python）

102. 二叉树的层序遍历

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

示例：
二叉树：[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]]:
    # ！！此代码错误，输出少了root节点和同层其他节点的子节点
        # result=[]
		# temp=[]
		# if not root:
		#     return None
		# left=root.left
		# right=root.right
		# if left:
		#     temp.append(left.val)
		# if right:
		#     temp.append(right.val)
		# if temp:
		#     result.append(temp)
		# temp=[]
		# if self.levelOrder(left):
		#     temp.append(self.levelOrder(left))
		# if self.levelOrder(right):
		#     temp.append(self.levelOrder(right))
		# if temp:
		#     result.append(temp[0][0])
		# return result
		
        # 正确代码如下：https://leetcode-cn.com/problems/binary-tree-level-order-traversal/comments/394128
        # 构建两个数组分别按层序遍历存放节点和节点值
        ans, d = [], [root] if root else []
        while d:
            ans.append([t.val for t in d])
            # 双层循环简便写法
            d = [r for t in d for r in (t.left, t.right) if r]
        return ans

总结：

1. 双层循环简便写法，先写外层，逐层向内：[r for t in d for r in (t.left, t.right) if r]
2. collections 是 python 内建的一个集合模块，里面封装了许多集合类，其中队列相关的集合只有一个：deque。 deque 是双边队列（double-ended queue），具有队列和栈的性质，在 list 的基础上增加了移动、旋转和增删等。
3. 动态规划也可做。