这是树的第7篇算法,力扣链接。
给你二叉树的根节点
root
,返回其节点值 自底向上的层序遍历 。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:[[15,7],[9,20],[3]]
这道题和上一道题的区别是从底向上,那我们层级遍历的逻辑就要改成栈的逻辑存储数据。
栈的模拟逻辑有两种方法,一种是每次插入都使老数据后移:
func levelOrderBottom(root *TreeNode) [][]int {
var result [][]int
if root == nil {
return result
}
queue := []*TreeNode{root}
for len(queue) > 0 {
var newQueue []*TreeNode
var subResult []int
for _, node := range queue {
subResult = append(subResult, node.Val)
if node.Left != nil {
newQueue = append(newQueue, node.Left)
}
if node.Right != nil {
newQueue = append(newQueue, node.Right)
}
}
for i := 0; i < len(result); i++ {
result[i], subResult = subResult, result[i]
}
result = append(result, subResult)
queue = newQueue
}
return result
}
还有一种是让数组在最后统一异动len(result)个单位。
func levelOrderBottom(root *TreeNode) [][]int {
var result [][]int
if root == nil {
return result
}
queue := []*TreeNode{root}
for len(queue) > 0 {
var newQueue []*TreeNode
var subResult []int
for _, node := range queue {
subResult = append(subResult, node.Val)
if node.Left != nil {
newQueue = append(newQueue, node.Left)
}
if node.Right != nil {
newQueue = append(newQueue, node.Right)
}
}
result = append(result, subResult)
queue = newQueue
}
for i := 0; i < len(result)/2; i++ {
result[i], result[len(result)-i-1] = result[len(result)-i-1], result[i]
}
return result
}