把二叉树打印成多行

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

思路

该题和按照之字形打印二叉树差不多，需要每一层输出一行，也就是每一层需要输出一个list,那么需要两个队列进行合作，一个队列出队（保存上一层节点），一个队列入队（保存前一个队列的左右节点，即下一层节点），只不过该题和按照之字形二叉树打印的区别在于，之字形二叉树需要区分奇数层和偶数层，那么就需要两个队列保存左右节点的顺序不一样，上一个队列先保存左节点，再保存右节点；下一个队列先保存右节点，再保存左节点。

代码

import java.util.ArrayList;
import java.util.LinkedList;

public class Print1 {

    public class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }

    }

    ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();

        if(pRoot == null) return list;

        LinkedList<TreeNode> queue1 = new LinkedList<>();
        LinkedList<TreeNode> queue2 = new LinkedList<>();

        queue1.add(pRoot);

        while (!queue1.isEmpty() || !queue2.isEmpty()){
            if(!queue1.isEmpty()){
                ArrayList<Integer> subList = new ArrayList<>();
                while (!queue1.isEmpty()){

                    TreeNode curNode = queue1.pollFirst();
                    subList.add(curNode.val);

                    if(curNode.left != null){
                        queue2.add(curNode.left);
                    }
                    if(curNode.right != null){
                        queue2.add(curNode.right);
                    }
                }
                list.add(subList);
            }else {
                ArrayList<Integer> subList1 = new ArrayList<>();
                while (!queue2.isEmpty()){
                    TreeNode curNode = queue2.pollFirst();
                    subList1.add(curNode.val);

                    if(curNode.left != null){
                        queue1.add(curNode.left);
                    }
                    if(curNode.right != null){
                        queue1.add(curNode.right);
                    }
                }
                list.add(subList1);
            }

        }


        return list;
    }
}