- 题目
给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。 - 示例
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
- 代码实现
数据结构:栈;
currStack:存储当前层的元素;
nextStack:存储下一层的元素。
order:boolean类型,控制左,右孩子的加入顺序,实现锯齿形。
思路:
a.在弹出currentStack元素时,将该元素的左,右子孩子记录到nextStack中,直到currStack元素全部弹出。(即当前层元素全部输出)(注:左,右孩子的加入顺序由order变量控制)
b.交换currStack和nextStack,使currStack始终指向当前需要输出的层。
c.order取反
import java.util.*;
public class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root){
List<List<Integer>> lists = new ArrayList<>();
//存储当前层元素
Stack<TreeNode> currStack = new Stack<>();
//存储下一层元素
Stack<TreeNode> nextStack = new Stack<>();
if(root!=null){
currStack.add(root);
}
//控制锯齿形
boolean order = true;
while (!currStack.empty()){
List<Integer> list = new ArrayList<>();
//将当前层元素全部弹出
while (!currStack.empty()){
TreeNode top = currStack.pop();
list.add(top.val);
//将弹出元素的左右孩子记录到nextStack中
if(order){
if(top.left!=null){
nextStack.add(top.left);
}
if(top.right!=null){
nextStack.add(top.right);
}
}else{
if(top.right!=null){
nextStack.add(top.right);
}
if(top.left!=null){
nextStack.add(top.left);
}
}
}
//交换currStack和nextStack,使currStack始终指向当前层
currStack = nextStack;
nextStack = new Stack<>();
lists.add(list);
//order取反
order =!order;
}
return lists;
}
}
注:常规层次遍历使用队列实现