6-树-二叉树的层序遍历 II

这是树的第7篇算法,力扣链接。

给你二叉树的根节点 root ,返回其节点值 自底向上的层序遍历 。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

示例 1:

6-树-二叉树的层序遍历 II_第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
}

你可能感兴趣的:(白话算法,算法,数据结构)