题目描述(和下一题一样)
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
方法:通过Queue实现,加入一个reverse的判断,reverse == false -> true ->false -> true ->false -> true ->false -> true ->......
其实就是广度优先遍历的变种。
与第二十二题对应,以后广度优先遍历就用队列Queue来实现吧,因为广度优先遍历使用arraylist模拟一个队列来存储相应的TreeNode,还不如直接用队列Queue实现。
package test;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
class TreeNode {
int val;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
class Solution {
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> result=new ArrayList<>();
if(pRoot==null){
return result;
}
Queue<TreeNode> queue=new LinkedList<TreeNode>();//这里必须声明是TreeNode类型
queue.add(pRoot);//这里才能往队列里面放TreeNode类型的变量
boolean reverse=false;
while(!queue.isEmpty()){
ArrayList<Integer> list=new ArrayList<>();//list存放的是Integer类型的数
int size=queue.size();//必须在这里定义,因为下面的for循环queue.size()就变化了
for(int i=0;i<size;i++){
TreeNode temp = queue.poll();//删除第一个元素并返回
if(!reverse){
list.add(temp.val);
}else{
list.add(0,temp.val);//每次加到0的位置,就自动逆序了
}
if(temp.left!=null)
queue.add(temp.left);//Queue中add()和offer()都是用来向队列添加一个元素
if(temp.right!=null)
queue.add(temp.right);//在容量已满的情况下,add()方法会抛出IllegalStateException异常,offer()方法只会返回 false
}
result.add(list);
reverse=!reverse;
}
return result;
}
}
public class Main {
public static void main(String[] args) {
TreeNode head=new TreeNode(1);//849
TreeNode second=new TreeNode(2);//850
TreeNode three=new TreeNode(3);//851
TreeNode four=new TreeNode(4);//852
TreeNode five=new TreeNode(5);//853
TreeNode six=new TreeNode(6);//854
TreeNode seven=new TreeNode(7);//855
head.right=three;
head.left=second;
second.right=five;
second.left=four;
three.right=seven;
three.left=six;
Solution p = new Solution();
ArrayList<ArrayList<Integer>> list = p.Print(head);
System.out.println(list);
}
}