JZ60 把二叉树打印成多行

原题链接

描述

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

例如：
给定的二叉树是{1,2,3,#,#,4,5}，该二叉树多行打印层序遍历的结果是

[ [1], [2,3], [4,5] ]

示例1

输入：{1,2,3,#,#,4,5}
返回值：[[1],[2,3],[4,5]]

示例2

输入：{8,6,10,5,7,9,11}
返回值：[[8],[6,10],[5,7,9,11]]

示例3

输入：{1,2,3,4,5}
返回值：[[1],[2,3],[4,5]]

思路

还是层次遍历，不过是要把每一层的结点值放在一个数组中，最后返回一个二维数组。

解答

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Queue;

public class Solution {
    ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        if (pRoot == null) return new ArrayList<>();

        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        Queue<TreeNode> queue = new ArrayDeque<>();

        queue.add(pRoot);

        while (!queue.isEmpty()) {
            int s = queue.size();
            ArrayList<Integer> arrayList = new ArrayList<>();

            //同一排的
            while (s-- > 0){
                TreeNode remove = queue.remove();

                arrayList.add(remove.val);

                if (remove.left != null) queue.add(remove.left);
                if (remove.right != null) queue.add(remove.right);
            }
            res.add(arrayList);
        }
        return res;
    }
}