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

一、题目

题目链接：力扣

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

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

    3
   / \
  9  20
    /  \
   15   7

返回： 

[3,9,20,15,7]

提示：

节点总数 <= 1000

---

二、题解

1、思路

🐭 BFS层次遍历：

• 题目要求的二叉树的 从上至下 打印（即按层打印），又称为二叉树的 广度优先搜索（BFS）。
• BFS 通常借助 队列 的先入先出特性来实现。

算法流程：

1.特例处理： 当树的根节点为空，则直接返回空列表 [] ；
2.初始化： 打印结果列表 res = [] ，包含根节点的队列 queue = [root] ；
3.BFS 循环： 当队列 queue 为空时跳出；

• 1.出队： 队首元素出队，记为 node；
• 2.打印： 将 node.val 添加至列表 tmp 尾部；
• 3.添加子节点： 若 node 的左（右）子节点不为空，则将左（右）子节点加入队列 queue ；

4.返回值： 返回打印结果列表 res 即可。

2、代码实现

🐭 BFS层次遍历：

class Solution {
public:
    vector<int> levelOrder(TreeNode* root) {
        vector<int> res;
        if(root == NULL)return res;

        TreeNode* p = root;
        queue<TreeNode*> que;
        que.emplace(p);

        while(!que.empty())
        {
            TreeNode* temp = que.front();// 返回第一个元素的引用
            // 因为下面判断了，压栈的内容不会为空，这里可以无需判断
            if(temp != NULL)res.emplace_back(temp->val);// 访问
            que.pop();// 出队

            if(temp->left != NULL)que.emplace(temp->left);// 左孩子入队
            if(temp->right != NULL)que.emplace(temp->right);// 右孩子入队
        }
        
        return res;
    }
};

3、复杂度分析

🐭 BFS层次遍历：

时间复杂度 O(n)： n为二叉树的节点数量，即 BFS 需循环n 次。
空间复杂度 O(n)： 最差情况下，即当树为平衡二叉树时，最多有 n/2 个树节点同时在 queue 中，使用 O(n)大小的额外空间。

4、运行结果

🐭 BFS层次遍历：