剑指 Offer 32 - II. 从上到下打印二叉树 IIjava题解

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;
    }
}