mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-31 16:14:39 +08:00
126 lines
2.3 KiB
Markdown
126 lines
2.3 KiB
Markdown
# [103. Binary Tree Zigzag Level Order Traversal](https://leetcode.com/problems/binary-tree-zigzag-level-order-traversal/)
|
|
|
|
## 题目
|
|
|
|
Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).
|
|
|
|
**For Example**:
|
|
Given binary tree [3,9,20,null,null,15,7],
|
|
|
|
```
|
|
|
|
3
|
|
/ \
|
|
9 20
|
|
/ \
|
|
15 7
|
|
|
|
```
|
|
|
|
return its zigzag level order traversal as:
|
|
|
|
```
|
|
|
|
[
|
|
[3],
|
|
[20,9],
|
|
[15,7]
|
|
]
|
|
|
|
```
|
|
|
|
|
|
## 题目大意
|
|
|
|
按照 Z 字型层序遍历一棵树。
|
|
|
|
## 解题思路
|
|
|
|
- 按层序从上到下遍历一颗树,但是每一层的顺序是相互反转的,即上一层是从左往右,下一层就是从右往左,以此类推。用一个队列即可实现。
|
|
- 第 102 题和第 107 题都是按层序遍历的。
|
|
|
|
|
|
|
|
|
|
## 代码
|
|
|
|
```go
|
|
|
|
package leetcode
|
|
|
|
/**
|
|
* Definition for a binary tree node.
|
|
* type TreeNode struct {
|
|
* Val int
|
|
* Left *TreeNode
|
|
* Right *TreeNode
|
|
* }
|
|
*/
|
|
|
|
// 解法一
|
|
func zigzagLevelOrder(root *TreeNode) [][]int {
|
|
if root == nil {
|
|
return [][]int{}
|
|
}
|
|
queue := []*TreeNode{}
|
|
queue = append(queue, root)
|
|
curNum, nextLevelNum, res, tmp, curDir := 1, 0, [][]int{}, []int{}, 0
|
|
for len(queue) != 0 {
|
|
if curNum > 0 {
|
|
node := queue[0]
|
|
if node.Left != nil {
|
|
queue = append(queue, node.Left)
|
|
nextLevelNum++
|
|
}
|
|
if node.Right != nil {
|
|
queue = append(queue, node.Right)
|
|
nextLevelNum++
|
|
}
|
|
curNum--
|
|
tmp = append(tmp, node.Val)
|
|
queue = queue[1:]
|
|
}
|
|
if curNum == 0 {
|
|
if curDir == 1 {
|
|
for i, j := 0, len(tmp)-1; i < j; i, j = i+1, j-1 {
|
|
tmp[i], tmp[j] = tmp[j], tmp[i]
|
|
}
|
|
}
|
|
res = append(res, tmp)
|
|
curNum = nextLevelNum
|
|
nextLevelNum = 0
|
|
tmp = []int{}
|
|
if curDir == 0 {
|
|
curDir = 1
|
|
} else {
|
|
curDir = 0
|
|
}
|
|
}
|
|
}
|
|
return res
|
|
}
|
|
|
|
// 解法二 递归
|
|
func zigzagLevelOrder0(root *TreeNode) [][]int {
|
|
var res [][]int
|
|
search(root, 0, &res)
|
|
return res
|
|
}
|
|
|
|
func search(root *TreeNode, depth int, res *[][]int) {
|
|
if root == nil {
|
|
return
|
|
}
|
|
for len(*res) < depth+1 {
|
|
*res = append(*res, []int{})
|
|
}
|
|
if depth%2 == 0 {
|
|
(*res)[depth] = append((*res)[depth], root.Val)
|
|
} else {
|
|
(*res)[depth] = append([]int{root.Val}, (*res)[depth]...)
|
|
}
|
|
search(root.Left, depth+1, res)
|
|
search(root.Right, depth+1, res)
|
|
}
|
|
|
|
``` |