LeetCode 102. Binary Tree Level Order Traversal（Python）

最新推荐文章于 2022-02-20 23:20:19 发布

原创最新推荐文章于 2022-02-20 23:20:19 发布 · 358 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

LeetCode-P 专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种解决二叉树层次遍历问题的方法，并提供了一个详细的Python实现示例。通过使用队列思想，在每层的最后一个位置后添加一个特殊节点来判断是否到达该层末尾。

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

题目描述：
Given a binary tree, return the level order traversal of its nodes’ values. (ie, from left to right, level by level).

For example:
Given binary tree [3,9,20,null,null,15,7],

return its level order traversal as:

[    
 [3],   
 [9,20],   
 [15,7] 
]

思路：
可以运用队列的思想，其中在每一层的最后一个位置后添加一个end节点来判断是否已经到了这一层的末尾，当然python可以直接运用列表代替队列的实现。

AC代码：

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:
            return []
        res, ans = [], []
        q = [root]
        # end结点来判断一层是否已经走到了末尾
        end = TreeNode('#')
        q.append(end)
        while True:
            temp = q.pop(0)
            if len(q) != 0 and temp.val == '#':
                q.append(end)
                res.append(ans)
                ans = []
            elif temp.val == '#' and len(q) == 0:
                res.append(ans)
                break
            else:
                ans.append(temp.val)
                if temp.left:
                    q.append(temp.left)
                if temp.right:
                    q.append(temp.right)
        return res