Leetcode102:二叉树的层次遍历，DFS&&BFS

最新推荐文章于 2022-05-01 13:29:30 发布

pqlwb

最新推荐文章于 2022-05-01 13:29:30 发布

阅读量280

点赞数

文章标签： Leetcode python DFS

本文链接：https://blog.csdn.net/weixin_43193954/article/details/89655586

版权

102:二叉树的层次遍历

题目描述

给定一个二叉树，返回其按层次遍历的节点值。（即逐层地，从左到右访问所有节点）。
例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

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

解题思路

想到DFS和BFS，这题最好的方法就是BFS,记住其中模版就可以

解法

解法一BFS

class Solution(object):
    def levelOrder(self, root):
        if not root: return []
        queue = collections.deque()
        result = []
        queue.append(root)
        while queue:
            curlen_level = len(queue)
            cur_level = []
            for _ in range(curlen_level):
                note = queue.popleft()
                cur_level.append(note.val)
                if note.left: queue.append(note.left)
                if note.right: queue.append(note.right)
                
            result.append(cur_level)
        return result

解法二DFS

class Solution(object):
    def levelOrder(self, root):
        if not root: return []
        self.result = []
        self._dfs(root, 0)
        return self.result
    def _dfs(self, note, level):
        if not node: return
        
        if len(self.result) < level + 1:
            self.result.append([])
        
        self.result[level].append(note.val)
        self._dfs(note.left, level + 1)
        self._dfs(note.left, level+1)