思路:用一个队列和列表来实现。主要是两层循环,外层循环用来获取队列大小size以及初始化一个列表list(用来存储当前层的节点),内层循环用来依次从队列中从左到右的取出当前层节点放到list中,并将当前节点的左右子结点插入到队尾。每次从队列中取出一个节点的时候都要对size进行自减操作,直到当size==0时,也就是当前层所有的节点都从队列中取出后,再重新开始外层循环,计算下一层的结点数size,并初始化一个用来保存当前层节点的列表list,进行新一轮的循环,以此类推。直到最后队列为空的时候,说明当前层没有节点了,算法结束。
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(root==null)
return res;
Queue<TreeNode> q = new ArrayDeque<TreeNode>();
TreeNode left = null;
TreeNode right = null;
TreeNode temp = root;
q.add(root);
int size = 1;
while(true){
size = q.size();
ArrayList<Integer> inner = new ArrayList<Integer>();
while(size!=0){
temp = q.poll();
inner.add(temp.val);
size--;
left = temp.left;
right = temp.right;
if(left!=null)
q.add(left);
if(right!=null)
q.add(right);
}
res.add(inner);
if(q.size()==0)
break;
}
return res;
}