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

NingDream816

于 2024-07-16 23:19:33 发布

阅读量358

点赞数 12

分类专栏：剑指offer 文章标签：算法 leetcode

本文链接：https://blog.csdn.net/weixin_51332735/article/details/140478701

版权

剑指offer 专栏收录该内容

24 篇文章 1 订阅

订阅专栏

一

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

示例

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

[3,9,20,15,7]

题解

    vector<int> floorTravel(TreeNode* root) {
        if (root == nullptr) return {};
        vector<int> res;
        std::queue<TreeNode*> q;
        q.push(root);
        while (!q.empty()) {
            TreeNode* cur_pop = q.front();
            if (cur_pop->left) q.push(cur_pop->left);
            if (cur_pop->right) q.push(cur_pop->right);
            res.push_back(cur_pop->val);
            q.pop();
        }
        return res;
    }

二

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

示例

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

返回其层次遍历结果：

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

题解

与一不同之处在于，需要将每一层隔开？那么怎么进行隔开呢？一层一层的进行处理。

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

        std::queue<TreeNode*> q{{root}};

        while (!q.empty()) {
            vector<int> v;
            // 每次开始前需要先固定q.size()
            // int size = q.size();
            // for (int i = 0; i < size; i++) 也可以
            for (int n = q.size(); n; --n) {
                TreeNode* tmp = q.front();
                q.pop();
                v.push_back(tmp->val);

                if (tmp->left) q.push(tmp->left);
                if (tmp->right) q.push(tmp->right);
            }
            res.push_back(v);
        }
        return res;
    }
};