打印二叉树合集

力扣 面试题32 - I. 从上到下打印二叉树

public int[] levelOrder(TreeNode root) {
        if(root==null) return new int[0];
        Queue<TreeNode> queue=new LinkedList<>();
        queue.add(root);
        ArrayList<TreeNode> temp=new ArrayList<>();
        while(!queue.isEmpty()){
            TreeNode node=queue.poll();
            temp.add(node);
            if(node.left!=null) queue.add(node.left);
            if(node.right!=null) queue.add(node.right);
            
        }
        int[] res=new int[temp.size()];
        for(int i=0;i<temp.size();i++){
            res[i]=temp.get(i).val;
        }
        return res;
    }

面试题32 - II. 从上到下打印二叉树 II

public List<List<Integer>> levelOrder(TreeNode root){
        if(root==null) return new LinkedList<>();
        List<List<Integer>> res=new LinkedList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        List<Integer> temp;
        queue.add(root);
        while(!queue.isEmpty()){
            temp=new LinkedList<>();
            for(int i=queue.size();i>0;i--){//这里注意如果是I=0开始遍历，后面queue有变化的话就会导致循环的次数出错
                TreeNode node=queue.poll();
                temp.add(node.val);
                if(node.left!=null) queue.add(node.left);
                if(node.right!=null) queue.add(node.right);
            }
            res.add(temp);
        }
        return res;
    }

面试题32 - III. 从上到下打印二叉树 III

方法一：层序遍历 + 双端队列（奇偶层逻辑分离）

public List<List<Integer>> levelOrder(TreeNode root) {
        if(root==null) return new LinkedList<>();
        List<List<Integer>> res=new LinkedList<>();
        Deque<TreeNode> queue=new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()){
            List<Integer> temp=new LinkedList<>();
            for(int i=queue.size();i>0;i--){
                TreeNode node=queue.removeFirst();
                temp.add(node.val);
                if(node.left!=null) queue.addLast(node.left);
                if(node.right!=null) queue.addLast(node.right);
            }
            res.add(temp);
            //如果奇偶没有分开，每一轮都会检查queue是否为空，所以分开写后需要在中间检查一次，否则结果中会有空的list
            if(queue.isEmpty()) break;
            temp=new LinkedList<>();
            for(int i=queue.size();i>0;i--){
                TreeNode node=queue.removeLast();
                temp.add(node.val);
                if(node.right!=null) queue.addFirst(node.right);
                if(node.left!=null) queue.addFirst(node.left);
            }
            res.add(temp);
        }
        return res;
    }

方法二：层序遍历 + 倒序

public List<List<Integer>> levelOrder(TreeNode root) {
        if(root==null) return new LinkedList<>();
        List<List<Integer>> res=new LinkedList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()){
            List<Integer> temp=new LinkedList<>();
            for(int i=queue.size();i>0;i--){
                TreeNode node=queue.poll();
                temp.add(node.val);
                if(node.left!=null) queue.add(node.left);
                if(node.right!=null) queue.add(node.right);
            }
            if(res.size()%2==1) Collections.reverse(temp);//判断奇偶层
            res.add(temp);
        }
        return res;
    }