103. 二叉树的锯齿形层次遍历

package Stack;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

public class ZigzagLevelOrder_103 {
	/*
	 * 给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。
	 * 
	 * 例如： 给定二叉树 [3,9,20,null,null,15,7],
	 * 
	 * 3 / \ 9 20 / \ 15 7
	 * 
	 * 返回锯齿形层次遍历如下：
	 * 
	 * [ [3], [20,9], [15,7] ]
	 * 
	 * 来源：力扣（LeetCode）
	 * 链接：https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
	 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
	 */

	public class TreeNode {
		int val;
		TreeNode left;
		TreeNode right;

		TreeNode(int x) {
			val = x;
		}
	}

	// 思路：
	// 1、用两个栈交替存储(因为用一个栈判断一行结束比较麻烦)
	// 2、如果子节点不为空则入栈，偶数行从右至左入栈，奇数行从左至右入栈
	
	public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
		Stack<TreeNode> stack1 = new Stack<>();
		Stack<TreeNode> stack2 = new Stack<>();
		List<List<Integer>> res = new ArrayList<>();
		// 判断为空的情况
		if(root==null){
            return res;
        }
		int flag = 1;
		stack1.push(root);
		while (!stack1.isEmpty() || !stack2.isEmpty()) {
			List<Integer> list = new ArrayList<>();
			if (flag == 1) {
				while (!stack1.isEmpty()) {
					root = stack1.pop();
					list.add(root.val);
					if (root.left != null)
						stack2.push(root.left);
					if (root.right != null)
						stack2.push(root.right);
				}
				res.add(list);
				flag = 0;
			}

			if (flag == 0) {
				while (!stack2.isEmpty()) {
					root = stack2.pop();
					list.add(root.val);
					if (root.right != null)
						stack1.push(root.right);
					if (root.left != null)
						stack1.push(root.left);
				}

				res.add(list);
				flag = 1;
			}
		}

		return res;

	}

}