102. 二叉树的层序遍历 (Python 实现)

x33g5p2x  于2022-02-18 转载在 Python  
字(0.6k)|赞(0)|评价(0)|浏览(327)

题目:
给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。

示例 1:

输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]

示例 2:

输入:root = [1]
输出:1

示例 3:

输入:root = []
输出:[]

代码:

  1. # Definition for a binary tree node.
  2. # class TreeNode:
  3. # def __init__(self, val=0, left=None, right=None):
  4. # self.val = val
  5. # self.left = left
  6. # self.right = right
  7. class Solution:
  8. def levelOrder(self, root: TreeNode) -> List[List[int]]:
  9. if not root:
  10. return []
  11. queue, res = [], []
  12. queue.append(root)
  13. while queue:
  14. k = []
  15. for _ in range(len(queue)):
  16. temp = queue.pop(0)
  17. k.append(temp.val)
  18. if temp.left:
  19. queue.append(temp.left)
  20. if temp.right:
  21. queue.append(temp.right)
  22. res.append(k)
  23. return res

相关文章