103.M 二叉树的锯齿形层序遍历

题目

https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal/

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

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

    3
   / \ 
  9   20 
      / \ 
     15  7 

返回锯齿形层序遍历如下：

[
  [3],
  [20,9],
  [15,7]
]

解题思路

其实就是层次遍历，但是需要把偶数层的结果做一次逆序。

实现代码

type TreeNode struct {
	Val int
	Left *TreeNode
	Right *TreeNode
}

func zigzagLevelOrder(root *TreeNode) [][]int {
	if root == nil {
		return nil
	}
	queue := []*TreeNode{root}
	reverse := false
	var result [][]int
	for len(queue) > 0 {
		size := len(queue)
		row := make([]int, size, size)
		for i := 0; i < size; i++ {
			node := queue[0]
			queue = queue[1:]
			if reverse {
				row[size-i-1] = node.Val
			} else {
				row[i] = node.Val
			}

			if node.Left != nil {
				queue = append(queue, node.Left)
			}
			if node.Right != nil {
				queue = append(queue, node.Right)
			}
		}
		result = append(result, row)
		reverse = !reverse
	}
	return result
}