Leetcode-每日一题【剑指 Offer 32 - II. 从上到下打印二叉树 II】

题目

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
9 20
    / \
   15   7

返回其层次遍历结果：

[
[3],
[9,20],
[15,7]
]

提示：

节点总数 <= 1000

解题思路

1.题目要求我们从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。这道题与【剑指 Offer 32 - I. 从上到下打印二叉树】思路完全一致，大家可以先去学习一下。

2.唯一不同的就是需要将每一层打印到一行，那么我们就新建一个（List<List<Integer>> res = new ArrayList<>();）动态数组去存储元素，在每次while循环开始时，我们都需要统计出队列中元素的个数，因为在结束了一次 while 循环后，队列中存储的元素是同一层的元素，所以我们需要统计出个数，并且新建一个动态数组 cur 将一层的元素存储进去，并且等 cur 数组存储结束后，再将 cur 数组存入我们最终的 res 数组中。直到队列为空，打印也就结束了。

代码实现

class Solution {public List<List<Integer>> levelOrder(TreeNode root) {if(root == null){return new ArrayList<>();}Queue<TreeNode> queue = new LinkedList<>();List<List<Integer>> res = new ArrayList<>();queue.add(root);while(!queue.isEmpty()){int k = queue.size();List<Integer> cur = new ArrayList<>();for(int i = 0; i < k; i++){TreeNode t = queue.poll();cur.add(t.val);if(t.left != null) queue.add(t.left);if(t.right != null) queue.add(t.right);}res.add(cur);}return res;
}
}