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

这篇具有很好参考价值的文章主要介绍了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;
}
}