class Solution {
public List levelOrder(TreeNode root) {
if(root == null)return new ArrayList();
Queue queue = new LinkedList();
queue.add(root);
List res = new ArrayList();
List a = new ArrayList();
a.add(root.val);
res.add(a);
while(!queue.isEmpty()){
List tmp = new ArrayList();
for(int i = queue.size(); i > 0; i--) {
TreeNode node = queue.poll();
tmp.add(node.val);
if(node.left != null) queue.add(node.left);
if(node.right != null) queue.add(node.right);
}
res.add(tmp);
}
res.remove(0);
return res;
}
}
和上一道题答题类似,但不一样的是,上一道题遍历到一个节点直接往后加就可以,但是这道题得把每一层都分清楚。我在while循环里面创建一个tem的list,它来装每一层的值,然后把它交给外层的list。我要每一次循环都把一层的数据都放进队列,每一次循环都把队列清空。先把根节点放入队列,然后看队列的大小,把队列里的每个节点的值都放进tem,并且把队列中每个节点的左右子节点都放进队列,这里注意用的是
for(int i = queue.size(); i > 0; i--)
不能写成i++形式,因为for循环每进行一次队列的大小都在变化。我也不知道我的res的list第一个元素为什么是空,我把第一个元素删了直接返回。
|