力扣103锯齿层序遍历树
给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
思路:还是要多提醒自己,在下笔写代码之前,一定要看清楚题目,一定要人工明细题目的过程,一定要动笔画草稿画出普通过程再下笔!!!!!
这个题目可以从草稿演示里面看出来:
1.需要按照遍历次数交替先进左孩子或者先进右孩子,可以设置一个变量来标志交替的次数
2.出队的顺序是从队头,而入队的顺序是从队尾,因此不仅要掌握Stack,Queue对LinkedList的调用,还要掌握自由存取的方法。比如get(index),remove(index)。
3.在添加和删除队列中的元素时,size()是实时变化的,所以在for循环前,需要先记录下队列的长度,以保证每一层经历一次while循环。
下面看代码:
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
if(root == null) return new ArrayList<List<Integer>>();
boolean flag = true;//flag==0 时,从右往左,flag==1 时,从左往右
List<TreeNode> list = new LinkedList<>();
List<List<Integer>> ans = new ArrayList<>();
list.add(root);
while(!list.isEmpty()) {
int size = list.size();
ArrayList<Integer> temp = new ArrayList<>();
for(int i=size-1;i>=0&&flag;i--) {//从队尾进入新的一层
TreeNode last = list.get(i);
if(last.right != null) list.add(last.right);
if(last.left != null) list.add(last.left);
}
for(int i=size-1;i>=0&&!flag;i--) {
TreeNode last = list.get(i);
if(last.left != null) list.add(last.left);
if(last.right != null) list.add(last.right);
}
for(int i=0;i<size;i++) {
temp.add(list.remove(0).val);//从队头出队构成答案序列
}
flag = !flag;//交替先进左孩子或者右孩子
ans.add(temp);
}
return ans;
}
}