问题描述:
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路:
之字形打印其实与上一个按层次打印基本相同,唯一不同的是每一层的顺序不一样,所以在这里可以定义一个变量,打印一层之后将其与1异或,作为打印顺序的标志位。
代码实现:
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> resultList = new ArrayList<ArrayList<Integer>>();
if(pRoot==null) return resultList;
//创建一个队列用于保存需要打印的结点
Queue<TreeNode> nodeQueue = new LinkedList<TreeNode>();
int nextNum =0; //本层已经打印的数量
int toBePrint = 1; //本层需要打印的总数量
int isLeftToRight = 0;//用来判断方向
nodeQueue.add(pRoot);
ArrayList<Integer> result = new ArrayList<Integer>();
while(!nodeQueue.isEmpty()){
TreeNode curNode = nodeQueue.poll();
result.add(curNode.val);
nextNum++;
if(curNode.left!=null) nodeQueue.add(curNode.left);
if(curNode.right!=null) nodeQueue.add(curNode.right);
//判断本层是否打印完毕
if(nextNum == toBePrint){
if(isLeftToRight==1){
resultList.add(reverse(result));
}else{
resultList.add(result);
}
toBePrint = nodeQueue.size();
nextNum = 0;
result = new ArrayList<Integer>();
isLeftToRight = isLeftToRight^1;
}
}
return resultList;
}
private ArrayList<Integer> reverse(ArrayList<Integer> result){
ArrayList<Integer> reverseResult = new ArrayList<Integer>();
for(int i=result.size()-1;i>=0;i--){
reverseResult.add(result.get(i));
}
return reverseResult;
}
}