面试题32-II.从上到下打印二叉树II

题目

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

示例

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

 3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

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

代码

和前一题稍稍有点不一样的，要将树中每一行的结果输出到一行
从上到下打印二叉树，又称为二叉树的广度优先搜索(BFS)，BFS通常借助队列来实现。
算法流程：

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

/**
 * 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) {
    	//判空
        if (root == null) return new ArrayList<>();
        //队列
        Deque<TreeNode> queue = new ArrayDeque<>();
        queue.addLast(root);//将root加入到队列
        //返回值
        List<List<Integer>> list = new ArrayList<>();
        
        while (queue.size() != 0) {
        	//内层的List
            List<Integer> innerList = new ArrayList<>();
            //前一个循环结束后，队列的长度就是树的下一层节点个数
            for (int i = queue.size(); i > 0; i--) {
                TreeNode node = queue.removeFirst();
                innerList.add(node.val);
                if (node.left != null) queue.addLast(node.left);
                if (node.right != null) queue.addLast(node.right);
            }

            list.add(innerList);
        }

        return list;
    }
}

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof