【算法笔记】大厂笔试/蓝桥杯/ICPC必学莫队算法

最新推荐文章于 2025-08-23 19:24:37 发布

原创最新推荐文章于 2025-08-23 19:24:37 发布 · 1k 阅读

15 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#算法 #笔记 #蓝桥杯

算法笔记专栏收录该内容

16 篇文章

订阅专栏

一、前言

在处理一类典型的离线区间查询问题时，我们常常面临这样一个选择：是用树状数组？线段树？还是其他暴力手段？这些方法在频繁修改或大规模数据下常常显得力不从心。此时，一种非常高效的“玄学”算法浮出水面：莫队算法。

二、莫队算法简介

1. 适用场景

莫队算法是一种离线处理区间查询的优化算法，适用于以下场景：

给定一个静态数组（不修改）
多次查询某个区间的某种统计信息
查询结果可以通过左右端点的微调（+1/-1）高效更新

2. 基本思想

莫队算法的核心在于：将所有查询排序，使得相邻查询的修改代价尽量小。它通常按以下方式排序所有查询：

将整个数组分块（设每块大小为 $n\sqrt n$ ）
将查询以左端点所属块编号为第一关键字，右端点为第二关键字进行排序
然后模拟左右指针从一个查询“滑”到下一个，边移动边更新答案

这种策略大大减少了查询之间的冗余计算，复杂度从 $O (n q)$ 降到 $\sqrt n$

三、经典应用：区间颜色种类查询

题目描述

给定一个长度为 $n$ 的数组，每个位置的值表示某种颜色，共有 $q$ 次查询，每次查询一个区间 $[l, r]$ ，回答这个区间内有多少种不同的颜色。

示例

输入：

n = 5, a = [1, 2, 1, 3, 2]
q = 3
query = [(1, 3), (2, 5), (1, 5)]

输出：

2
3
3

四、莫队算法实现步骤

Step 1: 读入所有查询，编号，离线存储

我们把每个查询记录成三元组 (l, r, id)，用于排序与结果回填。

Step 2: 按莫队排序规则排序所有查询

int block_size = sqrt(n);
sort(queries.begin(), queries.end(), [&](const Query& a, const Query& b) {
    int block_a = a.l / block_size;
    int block_b = b.l / block_size;
    if (block_a != block_b) return block_a < block_b;
    return a.r < b.r;
});

Step 3: 使用双指针移动并维护答案

用一个计数数组 cnt[color] 来维护每种颜色的出现次数，使用变量 res 来维护当前不同颜色的数量。

void add(int x) {
    if (++cnt[a[x]] == 1) ++res;
}

void remove(int x) {
    if (--cnt[a[x]] == 0) --res;
}

Step 4: 枚举每个查询

int l = 0, r = -1;
for (auto q : queries) {
    while (l > q.l) add(--l);
    while (r < q.r) add(++r);
    while (l < q.l) remove(l++);
    while (r > q.r) remove(r--);
    ans[q.id] = res;
}

五、完整代码示例（C++）

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 10;
int a[N], cnt[N], ans[N];
int res = 0;

struct Query {
    int l, r, id;
};

int block_size;

bool cmp(const Query& x, const Query& y) {
    int bx = x.l / block_size, by = y.l / block_size;
    if (bx != by) return bx < by;
    return x.r < y.r;
}

void add(int x) {
    if (++cnt[a[x]] == 1) ++res;
}

void remove(int x) {
    if (--cnt[a[x]] == 0) --res;
}

int main() {
    int n, q;
    cin >> n >> q;
    block_size = sqrt(n);
    for (int i = 0; i < n; ++i) cin >> a[i];

    vector<Query> queries(q);
    for (int i = 0; i < q; ++i) {
        cin >> queries[i].l >> queries[i].r;
        --queries[i].l, --queries[i].r; // 转为0-indexed
        queries[i].id = i;
    }

    sort(queries.begin(), queries.end(), cmp);

    int l = 0, r = -1;
    for (auto q : queries) {
        while (l > q.l) add(--l);
        while (r < q.r) add(++r);
        while (l < q.l) remove(l++);
        while (r > q.r) remove(r--);
        ans[q.id] = res;
    }

    for (int i = 0; i < q; ++i) cout << ans[i] << '\n';

    return 0;
}