题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
public class Solution {
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> res=new ArrayList<>();
if(pRoot==null)
return res;
LinkedList<TreeNode> queue=new LinkedList<>();
queue.offer(pRoot);
ArrayList<Integer> list = new ArrayList<>();
int start=0;
int end=1;
while(!queue.isEmpty())
{
TreeNode node=queue.pop();
list.add(node.val);
start++;
if(node.left!=null)
{
queue.offer(node.left);
}
if(node.right!=null)
{
queue.offer(node.right);
}
if(start==end)
{
start=0;
end=queue.size();
res.add(new ArrayList<>(list));
list.clear();
}
}
return res;
}
}