请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.Stack;
import java.util.ArrayList;
public class Solution {
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> list=new ArrayList<ArrayList<Integer>>();
if(pRoot==null)
return list;
Stack<TreeNode> oddStack=new Stack<>();
Stack<TreeNode> evenStack=new Stack<>();
oddStack.add(pRoot);
boolean flag=true;
while(!(oddStack.isEmpty()&&evenStack.isEmpty())){
ArrayList<Integer> temp=new ArrayList<>();
if(flag==true){//奇数行,odd中元素节点的孩子按先左孩子孩子后右孩子顺序入even
while(!oddStack.isEmpty()){
TreeNode t=oddStack.peek();
temp.add(t.val);
if(t.left!=null){
evenStack.add(t.left);
}
if(t.right!=null){
evenStack.add(t.right);
}
oddStack.pop();
}
list.add(temp);
flag=false;
}else{//偶数行,even中元素节点的孩子按先右孩子孩子后左孩子顺序入odd
while(!evenStack.isEmpty()){
TreeNode t=evenStack.peek();
temp.add(t.val);
if(t.right!=null){
oddStack.add(t.right);
}
if(t.left!=null){
oddStack.add(t.left);
}
evenStack.pop();
}
list.add(temp);
flag=true;
}
}
return list;
}
}