1.二刷
代码
重点是,如何确定打印一层结点。这里通过queue.size实现
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root==null) return new LinkedList<List<Integer>>();
List<List<Integer>> res=new ArrayList<>();//存放结果
LinkedList<TreeNode> queue=new LinkedList<>();//存放当前队列
queue.addLast(root);//将根节点加入队列
List<Integer> tmp;//暂时存放每一层结果
TreeNode before;//存放队列弹出的结点
while(!queue.isEmpty()){ //队列不为空
tmp=new ArrayList<>();//存放本层结果
int size=queue.size();//循环queue.size次,就遍历了本层结点
for(int i=0;i<size;i++){
before=queue.removeFirst();
tmp.add(before.val);//将头结点加入结果集
if(before.left!=null)
queue.addLast(before.left);
if(before.right!=null)
queue.addLast(before.right);
}
res.add(tmp);
}
return res;
}
}
复杂度
时间复杂度O(N):N为二叉树的结点数量
空间复杂度O(N):最差情况下,即当树为平衡二叉树时,最多有 N/2 个树节点同时在 queue 中,使用 O(N) 大小的额外空间。
结果
内存消耗还是很大
2.一刷
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root==null)
return new ArrayList<>();
LinkedList<TreeNode> queue=new LinkedList<>();
List<List<Integer>> res=new ArrayList<>();
TreeNode before;
queue.add(root);
while(!queue.isEmpty()){
ArrayList<Integer> tmp=new ArrayList<>();
//当前层
int size=queue.size();
for(int k=0;k<size;k++){
before=queue.poll();
tmp.add(before.val);
if(before.left!=null)
queue.add(before.left);
if(before.right!=null)
queue.add(before.right);
}
res.add(tmp);
}
return res;
}
}