锯齿形层次遍历题目
给你二叉树的根节点 root
,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
题目分析:
- 首先根节点1加入队列 反转标识为false
- 第二层 反转标识为true 此时当前指针指向队尾3
并且将3加入答案数组同时指针减1 指数字2 将2 加入数组 - 第三层和层次遍历相同的逻辑 反转标识为ture 依次将数字加入队列
相比于二叉树的层次遍历,只是从二叉树的第二层开始变化了遍历的顺序。 只需要在往答案数组添加答案时, 改变队列中的顺序。
代码实现:
class Solution {
public static int max = 2001;
public static TreeNode[] queue = new TreeNode[max];
public static int l,r;
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> ans = new ArrayList<>();
// false 代表从左往右
// true 代表从右往左
Boolean isReverse =false;
if (root!=null){
l=r=0;
queue[r++] = root;
while (l<r){
List<Integer> level = new ArrayList<>();
int size = r-l;
// reverse == false, 左 -> 右, l....r-1, 收集size个
// reverse == true, 右 -> 左, r-1....l, 收集size个
// 左 -> 右, i = i + 1
// 右 -> 左, i = i - 1
for (int i = isReverse?r-1:l,j = isReverse? -1:1,k = 0;
k<size;k++,i = i+j){
TreeNode cur = queue[i];
level.add(cur.val);
}
for (int i = 0; i < size; i++) {
TreeNode cur = queue[l++];
if (cur.left !=null){
queue[r++] = cur.left;
}
if (cur.right !=null){
queue[r++] = cur.right;
}
}
ans.add(level);
isReverse = !isReverse;
}
}
return ans;
}
}