题目:请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
解析:在层次遍历二叉树的基础上改变不同行的打印顺序
①层次遍历二叉树;
②在层次遍历二叉树的基础上对层数进行判断该层是奇数层还是偶数层;
③奇数层的结点可以直接打印,而偶数层的结点则将其添加到栈中,再出栈打印即为反序。
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> List = new ArrayList<ArrayList<Integer>>();
if (pRoot == null) return List;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(pRoot);
int count = 1;
while(!queue.isEmpty()){
int size = queue.size();
ArrayList<Integer> list = new ArrayList<Integer>();
Stack<Integer> stack = new Stack<Integer>();
//对每一层(count代表层数)的结点进行打印,同时将下一层的结点添加进队列中
while (size!=0){
TreeNode node = queue.poll();
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
//奇数层的树节点从左到右输出
if (count%2!=0){
list.add(node.val);
}else{
//偶数层的结点压入栈中
stack.push(node.val);
}
size--;
}
//偶数层的结点出栈打印-----变成反序
if (count%2==0){
while(!stack.isEmpty())
list.add(stack.pop());
}
//将每层的结点全部添加到List中
List.add(list);
count++;
}
return List;
}