102 二叉树层序遍历

全栈程序员站长

发布于 2022-08-31 11:30:57

1900

发布于 2022-08-31 11:30:57

文章被收录于专栏：全栈程序员必看全栈程序员必看

大家好，又见面了，我是你们的朋友全栈君。

层序遍历，每次层的输出是是一个一维数组，整个二叉树的输出结果是二维数组

BFS遍历，依托于队列结构，每次在根节点出栈的时候，将其值加在结果列表中，然后将他的左右孩子节点入队列。层序遍历相对于BFS，需要知道每一层有多少个节点。

因此，我们需要稍微修改一下代码，在每一层遍历开始前，先记录队列中的结点数量 nn（也就是这一层的结点数量），然后一口气处理完这一层的 n 个结点。参考

class Solution:
    def levelOrder(self, root: TreeNode):
        r = []
        if not root:return r
        quee = [root]
        while quee:
            temp = []
            k = len(quee)# 当前层右多少个节点
            for i in range(k):
                root = quee.pop(0)
                temp.append(root.val)
                if root.left:quee.append(root.left)
                if root.right:quee.append(root.right)

            r.append(temp)
        return r

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/143428.html原文链接：https://javaforall.cn

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2022年5月2，如有侵权请联系 cloudcommunity@tencent.com 删除

java