从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
public class Solution {
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> re=new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> temp=new ArrayList<>();
Queue<TreeNode> q=new LinkedList<TreeNode>();
if(pRoot==null)
return re;
int next=0,now=1;//next:代表下一层结点数 now:代表当前层的结点数
q.add(pRoot);
while(!q.isEmpty()){
TreeNode t=q.remove();
now--;
temp.add(t.val);
if(t.left!=null){
q.add(t.left);
next++;
}
if(t.right!=null){
q.add(t.right);
next++;
}
if(now==0){
now=next;
next=0;
re.add(new ArrayList<Integer>(temp));
temp.clear();
}
}
return re;
}
}