LeetCode刷题进阶之从上到下打印二叉树(剑指Offer 32 - I)

一、题目

演示示例：

二、测试代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int[] levelOrder(TreeNode root) {
        List<Integer> list=new ArrayList<>();
        int[] res={};
        if(root==null){
            return res;
        }
        Queue<TreeNode> queue=new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            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=new int[list.size()];
        for(int i=0;i<list.size();i++){
            res[i]=list.get(i);
        }
    return res;
    }
}

三、运行情况

四、刷题总结

本题的主要思路步骤为：

（1）特例处理： 当树的根节点为空，则直接返回空列表 [] ；
（2）初始化： 打印结果列表 res = [] ，包含根节点的队列queue = [root] ；
（3）BFS 循环： 当队列 queue 为空时跳出；
（4）出队： 队首元素出队，记为 temp；
（5）打印： 将 temp.val 添加至列表 list 尾部；
（6）添加子节点： 若 temp的左（右）子节点不为空，则将左（右）子节点加入队列 queue ；
（7）返回值： 返回打印结果列表 res 即可。

传送门：二叉树的层次遍历