LeetCode102. Binary Tree Level Order Traversal （思路及python解法）

最新推荐文章于 2025-02-03 15:37:10 发布

原创最新推荐文章于 2025-02-03 15:37:10 发布 · 294 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

179 篇文章

订阅专栏

Python

173 篇文章

订阅专栏

本文介绍了一种使用BFS算法实现的二叉树层序遍历方法，通过存储每一层的节点值并遍历下一层节点，最终返回按层级顺序排列的节点值列表。

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]
]

输出层序遍历的节点。

第一个想法就是BFS，一边存储每一层的节点，一边然后再利用这层节点遍历下一层节点。

其中tempval存储当前层的val，tempnode存储当前层的节点。

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        
        if root is None:return root
        bfs=[root]
        final=[]
        while len(bfs)!=0:
            tempval=[]
            tempnode=[]
            for node in bfs:
                tempval.append(node.val)
                if node.left:
                    tempnode.append(node.left)
                if node.right:
                    tempnode.append(node.right)
            final.append(tempval)
            bfs=tempnode
        return final