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

最新推荐文章于 2022-09-23 22:00:04 发布

船桥

最新推荐文章于 2022-09-23 22:00:04 发布

阅读量246

点赞数

分类专栏： Python LeetCode

本文链接：https://blog.csdn.net/AIpush/article/details/103230845

版权

LeetCode 同时被 2 个专栏收录

179 篇文章 0 订阅

订阅专栏

Python

173 篇文章 0 订阅

订阅专栏

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