leetcode-102-Binary-Tree-Level-Order-Traversal

发表于 | 更新于: | 分类于 |

1.题目

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

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

3

/ \
9 20
/ \
15 7
返回其层次遍历结果：

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

2.解

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        # DFS 深度优先搜索
        
        if not root:
            return []
        # res是一个二维数组
        self.res = []
        self._dfs(root,0)
        return self.res
    
    def _dfs(self,node,layer):
        # 给当前层分配[]，接下来将节点的值存在当前层的list中
        if len(self.res) < layer + 1:
            self.res.append([])
        self.res[layer].append(node.val)
        if node.left:
            self._dfs(node.left,layer+1)
        if node.right:
            self._dfs(node.right,layer+1)
			

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        # BFS 广度优先搜索
        # 从上到下，每一层的节点进入队列
        
        if not root:
            return []
        
        nodes = [root]
        res = []
        
        while nodes:
            next_nodes = []
            sub_res = []
            for node in nodes:
                sub_res.append(node.val)
                if node.left:
                    next_nodes.append(node.left)
                if node.right:
                    next_nodes.append(node.right)
            res.append(sub_res)
            nodes = next_nodes
        return res