剑指 Offer 32 - II. 从上到下打印二叉树 II

题目来源
剑指 Offer 32 - II. 从上到下打印二叉树 II

题目思路

• I. 按层打印： 题目要求的二叉树的 从上至下 打印（即按层打印），又称为二叉树的 广度优先搜索（BFS）。BFS 通常借助 队列 的先入先出特性来实现。
• II. 每层打印到一行： 将本层全部节点打印到一行，并将下一层全部节点加入队列，以此类推，即可分为多行打印。

BFS

• 特例处理： 当根节点为空，则返回空列表 [] ；
• 初始化： 打印结果列表 result = [] ，包含根节点的队列 queue.add(root) ；
• 新建一个临时列表 list，用于存储当前层打印结果；
• BFS 循环： 当队列 queue 为空时跳出；
  • 当前层打印循环： 循环次数为当前层节点数（即队列 queue 长度）；
  • 出队： 队首元素出队，记为 node；
  • 打印： 将 node.val 添加至 list尾部；
  • 添加子节点： 若 node 的左（右）子节点不为空，则将左（右）子节点加入队列 queue ；
• 将当前层结果 list添加入 result。
• 删除list的元素
• 返回值： 返回打印结果列表 result 即可。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        List<Integer> list = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if(root != null) queue.add(root);
        while(!queue.isEmpty()){
            for(int i = queue.size();i>0;i--){
                TreeNode node = queue.poll();
                list.add(node.val);
                if(node.left != null) queue.add(node.left);
                if(node.right != null) queue.add(node.right);
            }
            result.add(new ArrayList(list));
            list.clear();
        }
        return result;
    }
}