题目:从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
题解:用ArrayList代替队列,使用start和end变量来作为判断每一层结点是否被全部添加至数组中。
import java.util.ArrayList;
/*
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> > list = new ArrayList<>();
if(pRoot==null){
return list;
}
ArrayList<TreeNode> queue = new ArrayList<>();
ArrayList<Integer> temp = new ArrayList<>();
queue.add(pRoot);
int start = 0,end = 1;
while(!queue.isEmpty()){
TreeNode pNode = queue.remove(0);
start++;
temp.add(pNode.val);
if(pNode.left!=null){
queue.add(pNode.left);
}
if(pNode.right!=null){
queue.add(pNode.right);
}
if(start==end){
end = queue.size();
start = 0;
list.add(temp);
temp = new ArrayList<Integer>();
}
}
return list;
}
}