leetcode第102题:二叉树的层序遍历
https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
【题目】
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。(即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
【思路】
使用队列,只要队列不为空,则遍历队列元素,并将孩子节点加入新队列中,遍历结束后,将新队列复制给原始的队列。
【代码】
python版本
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
queue = [root]
res = []
# 只要不为空,就得继续遍历
while len(queue) > 0:
tmp = []
res_tmp = []
# 遍历所有元素
while len(queue) > 0:
node = queue.pop(0)
if not node:
continue
res_tmp.append(node.val)
tmp.append(node.left)
tmp.append(node.right)
if len(res_tmp) > 0:
queue = tmp
res.append(res_tmp)
return res
【相似题目】
103. 二叉树的锯齿形层次遍历
解题思路:层次遍历,对偶数层的结果逆序。
107. 二叉树的层次遍历 II
解题思路:层次遍历+结果逆序。
429. N叉树的层序遍历
解题思路:层次遍历。
637. 二叉树的层平均值
解题思路:层次遍历+对每一层的所有元素取平均值。