60把二叉树打印成多行

题目描述

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

思路分析

相当于层次打印，用一个队列作为辅助。在每层结束后，处理一下打印输出。

代码实现

public static ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
    Queue<TreeNode> queue = new LinkedList<>();
    ArrayList<ArrayList<Integer>> lists = new ArrayList<>();

    queue.add(pRoot);
    while (!queue.isEmpty()){
        ArrayList<Integer> arrayList = new ArrayList<>();
        //count为每一层的结点个数
        int count = queue.size();
        while (count-->0){
            TreeNode node =queue.poll();
            if(node==null){
                continue;
            }
            arrayList.add(node.val);
            queue.add(node.left);
            queue.add(node.right);
        }
        if(arrayList.size()!=0){
            lists.add(arrayList);
        }
    }
    return lists;
}