RMQ算法c++模板

呵呵HK

于 2024-07-03 22:43:54 发布

阅读量414

点赞数 8

分类专栏：题解文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/u010339252/article/details/140164033

版权

题解专栏收录该内容

15 篇文章 0 订阅

订阅专栏

RMQ（Range Minimum Query）问题是在一个数组中查找给定区间内最小元素的问题。它是一个非常常见的数据结构问题，有很多高效的算法可以解决这个问题，例如 $Sparse Table$ 算法和Segment Tree算法。下面，我将详细介绍 $Segment Tree$ 算法的C++实现，因为它是一种非常强大且易于理解的解决方案。

$Segment Tree$ 是一种二叉树形数据结构，它允许快速查询数组中的最小值、最大值、总和等统计信息。 $Segment Tree$ 的每个节点都包含了区间的信息，并且它的左子节点和右子节点分别包含区间和的信息。

以下是Segment Tree的C++实现：

#include <iostream>
#include <vector>
#include <climits> // 用于INT_MAX
#include <algorithm> // 用于std::min

// 定义一个Segment Tree节点结构
struct SegmentTreeNode {
    int start, end, min;
    SegmentTreeNode *left, *right;
    SegmentTreeNode(int s, int e) : start(s), end(e), min(INT_MAX), left(nullptr), right(nullptr) {}
};

// 构建Segment Tree
SegmentTreeNode* buildTree(const std::vector<int>& nums, int start, int end) {
    if (start > end) return nullptr;
    SegmentTreeNode* root = new SegmentTreeNode(start, end);
    if (start == end) {
        root->min = nums[start];
        return root;
    }
    int mid = start + (end - start) / 2;
    root->left = buildTree(nums, start, mid);
    root->right = buildTree(nums, mid + 1, end);
    root->min = std::min(root->left->min, root->right->min);
    return root;
}

// 查询区间最小值
int queryMin(SegmentTreeNode* root, int start, int end) {
    if (root == nullptr || start > end || root->start > end || root->end < start) return INT_MAX;
    if (root->start >= start && root->end <= end) return root->min;
    return std::min(queryMin(root->left, start, end), queryMin(root->right, start, end));
}

// 更新区间内的值（这里不展示，因为RMQ问题通常不需要更新）

// 销毁Segment Tree
void destroyTree(SegmentTreeNode* root) {
    if (root == nullptr) return;
    destroyTree(root->left);
    destroyTree(root->right);
    delete root;
}

int main() {
    std::vector<int> nums = {1, 3, 2, 7, 9, 11};
    SegmentTreeNode* root = buildTree(nums, 0, nums.size() - 1);
    std::cout << "Minimum in range [1, 3]: " << queryMin(root, 1, 3) << std::endl;
    std::cout << "Minimum in range [0, 5]: " << queryMin(root, 0, 5) << std::endl;
    destroyTree(root);
    return 0;
}

在这个实现中，我们首先定义了一个结构体来表示 $Segment Tree$ 的节点。函数用于从给定的数组构建 $Segment Tree$ 。函数用于查询区间内的最小值。请注意，这个实现假设数组是从0开始索引的。