解题思路-leetcode102题:二叉树的层序遍历
题目描述:
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:本题利用双边队列解题,首先将二叉树加入双边队列中,然后取出第一个元素,即二叉树第一层,然后新建一个列表tmp,遍历取出的数据,先遍历根节点,再遍历左孩子,再遍历右孩子,存在则加入列表。然后依次取出队列中的元素,直至遍历完成,最后返回存储所有tmp的列表res。代码如下:
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if not root:
return []
res = []
nums = collections.deque()
nums.append(root)
print(nums)
while nums:
tmp = []
for i in range(len(nums)):
node = nums.popleft()
# print(node)
tmp.append(node.val)
if node.left:
nums.append(node.left)
if node.right:
nums.append(node.right)
res.append(tmp)
return res
提交后,通过。