Day. 27 Binary Tree Level Order Traversal

Leetcode #102. Binary Tree Level Order Traversal

简单来说我们要回传每一层树的值
ex:

    3
   / \
  2   10
 /   /  \
1   8   13

output: [[3],[2,10],[1,8,13]]

防雷
防雷
防雷

大家还记得preOrder吗？
preOrder是从上到下的走访，我们只要在DFS的时候，多带一个层数就可以解掉这题了～

程序:

func levelOrder(root *TreeNode) [][]int {
	result := [][]int{}
	if root == nil {
		return result
	}

	DFS(&result, root, 0)

	return result
}

func DFS(result *[][]int, node *TreeNode, level int) {
	if node == nil {
		return
	}

    // 把node的值 加到slice里面
	if len(*result) <= level {
		*result = append(*result, []int{node.Val})
	} else {
		(*result)[level] = append((*result)[level], node.Val)
	}

    // 层数要+1
	DFS(result, node.Left, level+1)
	DFS(result, node.Right, level+1)
}