剑指offer 第6天搜索与回溯算法（简单）C++题解

最新推荐文章于 2024-08-09 20:36:57 发布

杨小木木

最新推荐文章于 2024-08-09 20:36:57 发布

阅读量63

点赞数

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

本文链接：https://blog.csdn.net/qq_39858171/article/details/127628977

版权

剑指offer 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

一、从上到下打印二叉树

剑指 Offer 32 - I. 从上到下打印二叉树https://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof/

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

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

3
/ \
9 20
/ \
15 7

[3,9,20,15,7]

提示：

节点总数 <= 1000

解题思路：

1.二叉树的层次遍历的特点为从上到下每层遍历，又称为广度优先遍历，通常借助队列来实现。

2.二叉树的深度遍历的特点为从上到下直至叶子节点，而称为深度优先遍历，通常借助栈来实现。

具体细节：

1.将根节点加入队列；

2.循环遍历队列元素，将左右孩子加入队列；

3.将值加入结果集。

复杂度分析：

1.时间复杂度：最多循环N次，O（N）。

2.空间复杂度：最差情况下有N/2个元素加入队列，故O（N）。

代码实现：

/**
 * 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> levelOrder(TreeNode* root) {
        queue<TreeNode*> q;
        vector<int> res;
        if (root) q.push(root);
        while (!q.empty()) {
            TreeNode *temp = q.front();
            q.pop();
            res.push_back(temp -> val);
            if (temp -> left) q.push(temp -> left);
            if (temp -> right) q.push(temp -> right);
        }
        return res;
    }
};

二、从上到下打印二叉树II

剑指 Offer 32 - II. 从上到下打印二叉树 IIhttps://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof/

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

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

3
/ \
9 20
/ \
15 7

返回其层次遍历结果：

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

提示：

节点总数 <= 1000

解题思路：

1.层次遍历优先想到用队列的先进先出特点实现。

2.每次循环遍历队列时将访问的元素加入数组中，并将下一层加入队列。

具体细节：

1.根节点入队；

2.循环遍历队列，新建一个temp数组用于记录当前层数据；

3.将当前层数据加入结果集中。

复杂度分析：

1.时间复杂度：O（N）。

2.空间复杂度：O（N）。

代码实现：

/**
 * 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<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        queue<TreeNode*> q;
        if (root) q.push(root);
        while (!q.empty()) {
            int n = q.size();
            vector<int> temp;
            for (int i = 0; i < n; i++) {
                TreeNode* t = q.front();
                q.pop();
                if (t -> left) q.push(t -> left);
                if (t -> right) q.push(t -> right);
                temp.push_back(t -> val);
            }
            res.push_back(temp);
        }
        return res;
    }
};

三、从上到下打印二叉树III

剑指 Offer 32 - III. 从上到下打印二叉树 IIIhttps://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof/

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

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

3
/ \
9 20
/ \
15 7

返回其层次遍历结果：

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

提示：

节点总数 <= 1000

解题思路：

1.解题思路与上两题一致，其中每层顺序按从左到右和从右到左交替打印。

2.在记录temp数组时通过标记交替进行从尾插入和从头插入。

具体细节：

1.根节点入队；初始化flag = true，表示从左到右，false为从右到左。

2.循环遍历队列，新建一个temp数组用于记录当前层数据；，flag为true从尾插入，为false从头插入。

3.将当前层数据加入结果集中。

复杂度分析：

1.时间复杂度：O（N）。

2.空间复杂度：O（N）。

代码实现：

/**
 * 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<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        queue<TreeNode*> q;
        bool flag = true;
        if (root) q.push(root);
        while (!q.empty()) {
            int n = q.size();
            vector<int> temp;
            for (int i = 0; i < n; i++) {
                TreeNode* t = q.front();
                q.pop();
                if (t -> left) q.push(t -> left);
                if (t -> right) q.push(t -> right);
                if (flag) temp.push_back(t -> val);
                else temp.insert(temp.begin(), t -> val);
            }
            flag = !flag;
            res.push_back(temp);
        }
        return res;
    }
};