LeetCode(力扣) 102题:二叉树的层序遍历----广度优先搜索(BFS)求解附带详细注释

题目描述
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

示例

输入:二叉树:[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循环),找到之后我们便在边界范围内遍历就好了。希望能帮助到大家,如果有疑惑的地方或者错误的地方,欢迎评论区滴滴我~

猜你喜欢

转载自blog.csdn.net/Just_do_myself/article/details/119811204
今日推荐