题目:请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
实现代码如下:
public static ArrayList<ArrayList<Integer> > PrintByZhi(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> arrs = new ArrayList<ArrayList<Integer> >();
if(pRoot == null)
{
return arrs;
}
//定义两个栈
Stack<TreeNode> stack1 = new Stack<TreeNode>();
Stack<TreeNode> stack2 = new Stack<TreeNode>();
stack1.push(pRoot);
TreeNode node = null;
while(!stack1.isEmpty() || !stack2.isEmpty())
{
ArrayList<Integer>list = new ArrayList<Integer>();
if(!stack1.isEmpty())
{
while(!stack1.isEmpty())
{
node = stack1.pop();
list.add(node.value);
if(node.leftNode != null)
{
stack2.push(node.leftNode);
}
if(node.rightNode != null)
{
stack2.push(node.rightNode);
}
}
arrs.add(list);
}
else
{
while(!stack2.isEmpty())
{
node = stack2.pop();
list.add(node.value);
if(node.rightNode != null)
{
stack1.push(node.rightNode);
}
if(node.leftNode!=null)
{
stack1.push(node.leftNode);
}
}
arrs.add(list);
}
}
return arrs;
}