题目描述

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

题解

大体框架与层次遍历相同，只是为了记录每层的节点数目，设置一个变量size，用于对当前层的节点进行遍历。

ArrayList<ArrayList<Integer>> Print(TreeNode root) {
    ArrayList<ArrayList<Integer>> res = new ArrayList<>();
    Queue<TreeNode> queue = new LinkedList<>();
    queue.offer(root);
    if (root == null)
        return res;

    while (!queue.isEmpty()) {
        int size = queue.size();
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = 0; i < size; i++) {
            TreeNode temp = queue.poll();
            list.add(temp.val);
            if (temp.left != null)
                queue.offer(temp.left);
            if (temp.right != null)
                queue.offer(temp.right);
        }
        res.add(list);
    }
    return res;
}