7-树-二叉树的锯齿形层序遍历

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

给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

示例 1:

7-树-二叉树的锯齿形层序遍历_第1张图片

输入:root = [3,9,20,null,null,15,7]
输出:[[3],[20,9],[15,7]]

这道题本质还是基础的层级遍历,不过唯一不同的是需要将奇数位的subQuery翻转一下。

func zigzagLevelOrder(root *TreeNode) [][]int {
	var result [][]int
	if root == nil {
		return result
	}
	queue, level := []*TreeNode{root}, 0
	for len(queue) > 0 {
		var newQueue []*TreeNode
		var subResult []int
		for _, node := range queue {
			if node.Left != nil {
				newQueue = append(newQueue, node.Left)
			}
			if node.Right != nil {
				newQueue = append(newQueue, node.Right)
			}
			subResult = append(subResult, node.Val)
		}
		if level%2 == 1 {
			for i, n := 0, len(subResult); i < n/2; i++ {
				subResult[i], subResult[n-1-i] = subResult[n-1-i], subResult[i]
			}
		}
		result = append(result, subResult)
		level++
		queue = newQueue
	}
	return result
}

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