LeetCode103.二叉树的锯齿形遍历Golang版

LeetCode103.二叉树的锯齿形遍历Golang版

1. 问题描述

给定一个二叉树，返回其节点值的锯齿形层序遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

例如：
给定二叉树 [3,9,20,null,null,15,7],

2. 思路

层序遍历模板题

3. 代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func zigzagLevelOrder(root *TreeNode) [][]int {
    var res [][]int
    if root == nil {
        return res
    }

    var layerCount int
    queue := []*TreeNode{root}

    for len(queue) > 0 {
        length := len(queue)
        var layer []int
        for i := 0; i < length; i++ {
            if queue[0].Left != nil {
                queue = append(queue, queue[0].Left)
            }
            if queue[0].Right != nil {
                queue = append(queue, queue[0].Right)
            }
            layer = append(layer, queue[0].Val)
            queue = queue[1:]
        }
        layerCount++
        if layerCount % 2 == 0 {
            layer = reverse(layer)
        }
        res = append(res, layer)
    }
    return res
}

func reverse(nums []int) []int {
    for i := 0; i < len(nums) / 2; i++ {
        nums[i], nums[len(nums) - 1 - i] = nums[len(nums) - 1 - i], nums[i]
    }
    return nums
}