编程语言
首页 > 编程语言> > 算法_二叉树_二叉树的层序遍历

算法_二叉树_二叉树的层序遍历

作者:互联网

文章目录

二叉树的层序遍历

leetcode链接

1.解法

之前说过,层序遍历就是广度优先搜索,实现广度优先搜索要使用队列,因为队列先进先出,符合一层一层遍历的逻辑,而是用栈先进后出适合模拟深度优先遍历也就是递归的逻辑。
在这里插入图片描述
先抛开这道题,我们只谈层序遍历。
具体步骤:

  1. 初始化队列,如果根节点不是空节点,就让根节点入队,否则直接返回空结果即可。
  2. 然后进入循环:首先将队首的节点的左孩子入队,然后将右孩子入队,之后将队首元素出队,这样就可以实现层序遍历了

补充:

  1. 如果要获得每层的节点个数,就像这道题一样,那么就在每次遍历之前统计一下队列里的元素个数,就是该层的节点个数。
  2. 注意在队首节点的左右孩子进队时,要判断该队首元素左右孩子是否为空,为空的话就不要入队了,不为空才入队。

代码如下:

def levelOrder(root):
    queue = [] # 用来层序遍历(广度优先搜索),元素为TreeNode
    result = [] # 存放结果,元素为list

    # 队列初始化,将根节点存入队列
    if root != None:
        queue.append(root)

    while queue:
        size = len(queue) # size是本层的元素个数,用来存放进临时list中
        re = [] # 存放本层的节点

        for i in range(size):
            # 队首节点的左孩子入队
            node = queue[0]
            if node.left != None: # 注意要判断是否为空
                queue.append(node.left)
            # 队首节点的右孩子入队
            if node.right != None:# 注意要判断是否为空
                queue.append(node.right)
            # 弹出队首节点,将值存入result中
            queue.pop(0)
            re.append(node.val)

        result.append(re) # 将本层结果存入result中

    return result

2.总结

算法

  1. 层序遍历(广搜)用队列,深搜用栈
  2. 层序遍历实现的一些细节:在上面的补充点里面都提到了。

3.题目推荐

102.二叉树的层序遍历
107.二叉树的层次遍历II
199.二叉树的右视图
637.二叉树的层平均值
429.N叉树的前序遍历
515.在每个树行中找最大值
116.填充每个节点的下一个右侧节点指针
117.填充每个节点的下一个右侧节点指针II
104.二叉树的最大深度
111.二叉树的最小深度

标签:遍历,层序,queue,入队,二叉树,节点
来源: https://blog.csdn.net/Dajian1040556534/article/details/121650560