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

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

题目描述

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

数据范围

树中节点的数量 [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 来做:


先判断根结点是否为空,如果不为空则推入队列。

根据队列中结点的个数来作为每次循环的条件。

每次将推出的结点的值放入答案,并判断其左右结点是否为空,不为空的话推入对应结点。

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



211cf37770954533b9d498eafae82dab.png


初始化: 先将根结点推入队列。


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


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

第三轮遍历:k = 4 ,更新 ans 数组并往队列推入弹出元素的孩子结点,发现已经没有可以推入的结点,直接返回 ans 数组即可。

f35c81d1002d4f2f859ffb319b71c50d.png

/**
 * 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;
    }
};


欢迎大家在评论区交流~

目录
相关文章
|
8月前
|
存储 Java C语言
剑指offer(牛客)——从尾到头打印链表
剑指offer(牛客)——从尾到头打印链表
45 1
|
8月前
|
算法
《剑指offer》之从上到下打印二叉树Ⅰ、Ⅱ、Ⅲ
《剑指offer》之从上到下打印二叉树Ⅰ、Ⅱ、Ⅲ
56 0
【剑指offer】-从上往下打印二叉树-22/67
【剑指offer】-从上往下打印二叉树-22/67
|
8月前
|
存储
【剑指offer】-把二叉树打印成多行-43/67
【剑指offer】-把二叉树打印成多行-43/67
|
8月前
|
存储
【剑指offer】- 按之字形顺序打印二叉树-45/67
【剑指offer】- 按之字形顺序打印二叉树-45/67
|
存储 算法 C语言
【C语言刷题】猜名次、猜凶手、杨辉三角、杨氏矩阵、字符串左旋、判断是否为左旋子串
【C语言刷题】猜名次、猜凶手、杨辉三角、杨氏矩阵、字符串左旋、判断是否为左旋子串
84 0
|
存储
剑指offer 32. 分行从上往下打印二叉树
剑指offer 32. 分行从上往下打印二叉树
47 0
剑指offer 33. 之字形打印二叉树
剑指offer 33. 之字形打印二叉树
84 0
剑指offer_二叉树---之字形打印二叉树
剑指offer_二叉树---之字形打印二叉树
62 0
力扣刷题记录——709. 转换成小写字母、771. 宝石与石头、704. 二分查找
力扣刷题记录——709. 转换成小写字母、771. 宝石与石头、704. 二分查找
117 0
力扣刷题记录——709. 转换成小写字母、771. 宝石与石头、704. 二分查找