107. 二叉树的层序遍历 II 中等
给定一个二叉树,返回其节点值自底向上的层序遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层序遍历为:
[
[15,7],
[9,20],
[3]
]
代码参考:
package main
import "fmt"
func main() {
tree := &Tree{}
tree.root = &TreeNode{Val: 3}
for _, v := range []int{9, 20, 15, 7} {
tree.BFSInsert(v)
}
fmt.Println(levelOrderBottom(tree.root))
}
func levelOrderBottom(root *TreeNode) [][]int {
if root == nil {
return nil
}
var res [][]int
queue := []*TreeNode{root}
for len(queue) > 0 {
var floor []int
level := len(queue)
for i := 0; i < level; i++ {
cur := queue[0]
queue = queue[1:]
floor = append(floor, cur.Val)
if cur.Left != nil {
queue = append(queue, cur.Left)
}
if cur.Right != nil {
queue = append(queue, cur.Right)
}
}
res = append(res, floor)
}
// 反转即可
for i, j := 0, len(res)-1; i < j; i, j = i+1, j-1 {
res[i], res[j] = res[j], res[i]
}
return res
}
最后编辑: kuteng 文档更新时间: 2021-06-05 10:16 作者:kuteng