求二叉树的层序遍历 python版本

金乌爬虫

已于 2022-10-12 16:52:39 修改

阅读量862

点赞数

分类专栏：面试题文章标签：面试

于 2020-11-22 09:49:55 首次发布

本文链接：https://blog.csdn.net/wanlitengfei/article/details/109921220

版权

面试题专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种实现二叉树层序遍历的方法，通过使用两个列表来跟踪当前层和下一层的节点，最终返回按层排列的节点值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个二叉树，返回该二叉树层序遍历的结果，（从左到右，一层一层地遍历）
例如：
给定的二叉树是{3,9,20,15,7},

该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]

示例1

输入: {1,2}

返回值: [[1],[2]]

示例2

输入: {1,2,3,4,5}

返回值: [[1],[2,3],[4,5]]

解题思路：重点是如何把在一层的节点放到一起，设置一个参数专门放一层的节点

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

#
# 
# @param root TreeNode类 
# @return int整型二维数组
#
class Solution:
    def levelOrder(self , root ):
        # write code here
        if not root:
            return []
        cur = [root]
        node = []
        res = []
        while cur:
            for i in cur:
                if i.left:
                    node.append(i.left)
                elif i.right:
                    node.append(i.right)
                res.append([i.val for i in cur])
                cur, node = node, []
        return res