题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路:
核心/考点:层级遍历二叉树、BFS。
变形:奇数层顺序遍历,偶数层逆序遍历。所以想清楚如何判断层性,以及如何逆序打印。
BFS(广度优先搜索):找到点与点之间的最短路径。先搜索距离最近的点,然后搜索次近的...直到找到目标点。
代码:
public class Solution {
public ArrayList > Print(TreeNode pRoot) {
//先建好已访问和将要访问的节点队列queue和存放打印的list
//if(pRoot==null) return null; 为啥加上这个通过率57.14% ???
LinkedList queue=new LinkedList<>();
ArrayList> lists=new ArrayList<>();
if(pRoot==null) return lists; //这样就可以,好像是返回null的问题?
boolean ceng=true;
//将根节点加入队列中
queue.add(pRoot);
//开始搜索
while(!queue.isEmpty()){
int size=queue.size();
//每一层新建一个list
ArrayList l=new ArrayList<>();
//每层进行遍历
//for(int i=0;i
//因为size时刻在变化!!每次循环里会往队列中添加节点!所以需要在循环前先找到层中节点数
for(int i=0;i
//将队列首节点弹出,并存放入list中
TreeNode node=queue.poll();
if(node==null){continue;} //特殊情况
//按照层数打印节点
if(ceng){
l.add(node.val);
}else{
l.add(0,node.val); //逆序打印
}
//将子节点加入队列中
queue.offer(node.left);
queue.offer(node.right);
}
//将该list放入总lists中
if(l.size()!=0) {lists.add(l);}
//!!!如何分辨奇数偶数?下一层就反转
ceng=!ceng;
}
return lists;
}
}