剑指offer 31. 不分行从上往下打印二叉树

Pandaconda

已于 2022-10-28 18:21:43 修改

阅读量112

点赞数

分类专栏： # 剑指offer第二版文章标签：算法数据结构剑指offer c++ 树

于 2022-08-31 18:21:12 首次发布

本文链接：https://blog.csdn.net/Newin2020/article/details/126629322

版权

剑指offer第二版专栏收录该内容

77 篇文章 8 订阅

订阅专栏

✍个人博客：https://blog.csdn.net/Newin2020?spm=1011.2415.3001.5343
📚专栏地址：剑指offer系列题解
📝原题地址：题目地址
📣专栏定位：为找工作的小伙伴整理常考算法题解，祝大家都能成功上岸！
❤️如果有收获的话，欢迎点赞👍收藏📁，您的支持就是我创作的最大动力💪

题目描述

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

数据范围

树中节点的数量 [0,1000]。

样例
输入如下图所示二叉树[8, 12, 2, null, null, 6, null, 4, null, null, null]
     8
    / \
   12  2
   /
  6
 /
4

输出：[8, 12, 2, 6, 4]

方法一：BFS O(n)

这道题的要求是需要我们按层来输出结点，那么可以用 BFS 来做：

先判断根结点是否为空，如果不为空则推入队列。
根据队列中结点的个数来作为每次循环的条件。
每次将推出的结点的值放入答案，并判断其左右结点是否为空，不为空的话推入对应结点。

我们举一个例子，来看看是如何实现的，假设我们要遍历这棵树：

在这里插入图片描述
初始化： 先将根结点推入队列。

在这里插入图片描述

第一轮遍历： k = 1 ，更新 ans 数组并往队列推入弹出元素的孩子结点。

在这里插入图片描述

第二轮遍历： k = 2 ，更新 ans 数组并往队列推入弹出元素的孩子结点。

在这里插入图片描述

第三轮遍历： k = 4 ，更新 ans 数组并往队列推入弹出元素的孩子结点，发现已经没有可以推入的结点，直接返回 ans 数组即可。
在这里插入图片描述

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> printFromTopToBottom(TreeNode* root) {
        if (!root)   return {};
        queue<TreeNode*> q;
        q.push(root);
        vector<int> ans;
        while (!q.empty())
        {
            int k = q.size();
            while (k--)
            {
                auto t = q.front();
                ans.push_back(t->val);
                q.pop();
                if (t->left != NULL)   q.push(t->left);
                if (t->right != NULL)  q.push(t->right);
            }
        }
        return ans;
    }
};