按之字形顺序打印二叉树

题目

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

思路

使用两个栈来储存相邻两层的节点，保证顺序

储存后的栈在下一层依次弹出节点，同时使用另一个栈保存再下一层的节点

public class Solution {
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> allList = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> list = new ArrayList<>();
TreeNode node = new TreeNode(0);
boolean flag = true;
if(pRoot==null)
return allList;
Stack<TreeNode> s1 = new Stack<TreeNode>();
s1.push(pRoot);
Stack<TreeNode> s2 = new Stack<TreeNode>();
while(!s1.empty()||!s2.empty()){
if(flag){
while(!s1.empty()){
node = s1.pop();
if(node!=null){
list.add(node.val);
s2.push(node.left);
s2.push(node.right);
}
}
}else{
while(!s2.empty()){
node = s2.pop();
if(node!=null){
list.add(node.val);
s1.push(node.right);
s1.push(node.left);
}
}
}

if(list.size()!=0){
allList.add(new ArrayList<Integer>(list));
list.clear();
flag = !flag;
}

}
return allList;

}

}