美文网首页
LeetCode 102 二叉树的层序遍历 (DFS方法)

LeetCode 102 二叉树的层序遍历 (DFS方法)

作者: JunfengsBlog | 来源:发表于2020-01-31 13:10 被阅读0次

对于二叉树的层序遍历,BFS方法是更为常用的思路。但我觉得用DFS递归的方法做也很好,下面贴出代码:

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        self.levelList = []
        if not root:
            return []
        self.dfs(root, 0)
        return self.levelList

    def dfs(self, root, level):
        if not root:
            return
        if len(self.levelList) < level + 1:
            self.levelList.append([])
        self.levelList[level].append(root.val)
        
        self.dfs(root.left, level+1)
        self.dfs(root.right, level+1)

相关文章

网友评论

      本文标题:LeetCode 102 二叉树的层序遍历 (DFS方法)

      本文链接:https://www.haomeiwen.com/subject/bvjgthtx.html