题目描述
给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[20,9],[15,7]]
示例 2:输入:root = [1]
输出:[[1]]
示例 3:输入:root = []
输出:[]提示:
树中节点数目在范围 [0, 2000] 内
-100 <= Node.val <= 100
做题思路
这个题目的意思就是从根节点开始。
逐层遍历二叉树,但是每一层遍历的顺序不能一致。
也就是说第一层从左往右遍历,第二层要从右往左遍历
然后最终要以集合的形式返回
这个题目我们不借助别的数据结构的话不好实现,所以对于本题我是借助了含有两个栈的数组来做了
我就是充分的利用了栈的特性,先进后出
但是仅仅用用一个栈实现还是不行的,因为有两种顺序需要我们遍历,从左到右。从右到左。
所以我定义了含两个栈的数组
然后就是我们可以定义一个变量 0/1 每次遍历完每层之后改变这个变量的值,就可以达到切换不同栈的功能
代码实现
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> res=new ArrayList<>();
if(root==null)return res;
Deque<TreeNode>[] stack=new Deque[2];
for(int i=0;i<2;i++){
stack[i]=new ArrayDeque<TreeNode>();
}
stack[1].push(root);
int flag=1;
while(!stack[flag].isEmpty()){
List<Integer> tem=new ArrayList<>();
int size=stack[flag].size();
for(int i=0;i<size;i++){
TreeNode head=stack[flag].pop();
if(flag==1){
if(head.left!=null){
stack[0].push(head.left);
}
if(head.right!=null){
stack[0].push(head.right);
}
}
if(flag==0){
if(head.right!=null){
stack[1].push(head.right);
}
if(head.left!=null){
stack[1].push(head.left);
}
}
tem.add(head.val);
}
res.add(tem);
flag=(flag+1)%2;
}
return res;
}
}