面试-二叉树的锯齿形层序遍历

题目

给你二叉树的根节点 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;
	}
}