题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
//打印当前结点的时候保存其左右孩子
public class Solution {
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
//按行保存
ArrayList<ArrayList<Integer> > nodesList=new ArrayList<ArrayList<Integer> >();
//保存每行的结点
ArrayList<Integer> list=new ArrayList<Integer>();
if(pRoot==null)
return nodesList;
//下一层的结点数
int nextLevelNodes=0;
//下一行要打印的结点数
int toBePrint=1;
Queue<TreeNode> queue=new LinkedList<TreeNode>();
queue.add(pRoot);
while(!queue.isEmpty()){
TreeNode tempNode=queue.poll();
list.add(tempNode.val);
if(tempNode.left!=null){
queue.add(tempNode.left);
++nextLevelNodes;
}
if(tempNode.right!=null){
queue.add(tempNode.right);
++nextLevelNodes;
}
--toBePrint;
//当前层结点打印全部后
if(toBePrint==0){
nodesList.add(list);
toBePrint=nextLevelNodes;
//清0,重新统计下一行的结点数
nextLevelNodes=0;
//清空集合
list=new ArrayList<Integer>();
}
}
return nodesList;
}
}