60.从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思路:
按层次输出二叉树
访问根节点,并将根节点入队。
当队列不空的时候,重复以下操作。
1、弹出一个元素。作为当前的根节点。
2、如果根节点有左孩子,访问左孩子,并将左孩子入队。
3、如果根节点有右孩子,访问右孩子,并将右孩子入队。
import java.util.ArrayList;
import java.util.LinkedList;
public class Solution {
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> listall =new ArrayList<>();//存放结果
if(pRoot==null){
return listall;
}
//使用队列,先进先出
LinkedList<TreeNode> queue=new LinkedList<>();
//存放每行的节点
ArrayList<Integer> list=new ArrayList<>();
//记录本层打印了多少个
int count=0;
int numall=1;//表示下一行节点总数
queue.add(pRoot);
while(!queue.isEmpty()){
TreeNode node=queue.remove();
//添加到本行的list集合中
list.add(node.val);
count++;
//每打印一个节点,就把此节点的下一层的左右节点加入队列,并记录下一层要打印的个数
if(node.left!=null){
queue.add(node.left);
}
if(node.right!=null){
queue.add(node.right);
}
//判断本层打印是否完成
if(count==numall){
//此时queue中存储的都是下一层的节点,则numall总个数即为queue大小
numall=queue.size();
count=0;
//把list添加到结果列表listall中。记录每一行
listall.add(list);
//重置list
list=new ArrayList();
}
}
return listall;
}
}