题目
给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
思路
相当于在层序遍历的基础之上,加了一个判断位,分别依次头插和尾插节点
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root){
List<List<Integer>> res = new ArrayList<List<Integer>>();
Queue<TreeNode> queue = new LinkedList<>();
if (root != null) {
queue.add(root);
}
boolean flag = false;
while(!queue.isEmpty()){
List<Integer> level = new ArrayList<>();
int n = queue.size();
while(n-- > 0){
TreeNode node = queue.poll();
if(node.left != null) queue.add(node.left);
if(node.right != null) queue.add(node.right);
if(flag){
level.add(0, node.val);//插入最前面,相当于头插法
} else {
level.add(node.val);//如果 index 没有传入实际参数,元素将追加至数组的最末尾。
}
}
flag =! flag;
res.add(level);
}
return res;
}
}