leetcode-107：二叉树的层序遍历 II

题目描述

给定一个二叉树，返回其节点值自底向上的层序遍历。 （即按从叶子节点所在层到根节点所在的层，逐层从左向右遍历）

解题思路

• 同二叉树的层序遍历，将最终结果反转即可

代码实现

func levelOrderBottom(root *TreeNode) [][]int {
	res := levelOrder(root)
	reverse(res)
	return res
}

func reverse(res [][]int) {
	for i, j := 0, len(res)-1; i < j; i, j = i+1, j-1 {
		res[i], res[j] = res[j], res[i]
	}
}

func levelOrder(root *TreeNode) [][]int {
	res := make([][]int, 0)
	if root == nil {
		return res
	}
	queue := make([]*TreeNode, 0)
	queue = append(queue, root)

	for len(queue) > 0 {
		list := make([]int, 0)
		length := len(queue)
		for i := 0; i < length; i++ {
			node := queue[0]
			queue = queue[1:]
			if node.Left != nil {
				queue = append(queue, node.Left)
			}
			if node.Right != nil {
				queue = append(queue, node.Right)
			}
			list = append(list, node.Val)
		}
		res = append(res, list)
	}
	return res
}