Leetcode 102. 二叉树的层次遍历(Python3)

102. 二叉树的层次遍历

给定一个二叉树，返回其按层次遍历的节点值。 （即逐层地，从左到右访问所有节点）。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

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

思路：

1.BFS

a.level  --> queue

b.Batch process

Time:O(N)

2.DFS

Time:O(N)

 

PS:一般写出广搜（人的思维）就可以了，如果还能写出深搜（机器的思维），会是面试的加分项

 

1.代码：

#BFS
class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:return []

        queue,res = [],[]
        queue.append(root)

        while queue:
            cl = []
            for _ in range(len(queue)):
                node = queue.pop(0)
                cl.append(node.val)
                if node.left:queue.append(node.left)
                if node.right:queue.append(node.right)
            res.append(cl)
        return res

for _ in range：在py中_ 下划线也可以当变量，在这里表示不使用该变量。

pop(0)：可以导入collections中的deque.leftpop()  来实现双端队列

 

2.代码：

#BFS
class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:return []
        self.res = []
        self.dfs(root,0)
        return self.res
    def dfs(self,node,level):
        if not node:return None
        if len(self.res) < level + 1:
            self.res.append([])
        self.res[level].append(node.val)
        self.dfs(node.left,level+1)
        self.dfs(node.right,level+1)