题目描述
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例
输入:二叉树:[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, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
#思考层序遍历需要设定两个停止条件,一个是到达最大深度时停止,另一个是到达最大宽度时停止
#树为空的情况,返回也为空
if root is None:
return []
#定义储存当前层节点以及所有节点值的列表
nodes, values = [], []
#首先由顶向下,先将第一层的根节点放入nodes
nodes.append(root)
#判断当前层没有节点,那么说明已经抵达最大深度,跳出循环
while nodes:
#储存当前层节点值
value = []
for node in nodes:
value.append(node.val)
values.append(value)
#浏览储存下一层节点
tmp = []
#浏览当前层节点的左右节点,如果有,则加入到下一层
for node in nodes:
if node.left:
tmp.append(node.left)
if node.right:
tmp.append(node.right)
#将下一层的节点赋给nodes,如果到了最后一层,tmp为空,跳出while循环,遍历结束
nodes = tmp
return values
运行结果
总结
这道算法题的难点主要是找到广度和深度的两个边界条件(即代码中的while循环和for循环),找到之后我们便在边界范围内遍历就好了。希望能帮助到大家,如果有疑惑的地方或者错误的地方,欢迎评论区滴滴我~