eg:基础莫队——求区间不同数字的个数
#include <bits/stdc++.h>#define int long longusing namespace std;const int N = 5e4+5, M = 2e5+5, S = 1e6+5;int n, m, len;//cnt[]表示当前询问区间内不同数的个数int w[N], cnt[S], ans[M];struct node {int id, l, r;}q[M];int get(int x) {return x / len;}bool cmp(node a, node b) {if (get(a.l) == get(b.l)) {//奇偶优化if (get(a.l) & 1) {return a.r < b.r;} else {return a.r > b.r;}}return get(a.l) < get(b.l);}void add(int x, int& res) {if (!cnt[x])res++;cnt[x]++;}void del(int x, int& res) {cnt[x]--;if (!cnt[x])res--;}signed main() {cin >> n;for (int i = 1; i <= n; i++)scanf("%lld", &w[i]);cin >> m;len = sqrt((double)n*n/m);for (int i = 1; i <= m; i++) {int l, r;scanf("%lld%lld", &l, &r);q[i] = {i, l, r};}sort(q+1, q+m+1, cmp);int i = 0, j = 1, res = 0;for (int k = 1; k <= m; k++) {int id = q[k].id, l = q[k].l, r = q[k].r;while (i < r) add(w[++i], res);while (i > r) del(w[i--], res);while (j < l) del(w[j++], res);while (j > l) add(w[--j], res);ans[id] = res;}for (int i = 1; i <= m; i++)cout << ans[i] << "\n";return 0;}
eg:带修改的莫队——单点修改,查询区间不同颜色数量
#include <bits/stdc++.h>
using namespace std;
const int N = 1e4+5, S = 1e6+5;
int n, m, mq, mc, len;
int w[N], cnt[S], ans[N];
struct query {
int id, l, r, t;
}q[N];
struct modify {
int p, c;
}c[N];
int get(int x) {
return x/len;
}
bool cmp(query a, query b) {
int al = get(a.l), ar = get(a.r);
int bl = get(b.l), br = get(b.r);
if (al != bl)
return al < bl;
if (ar != br)
return ar < br;
return a.t < b.t;
}
void add(int x, int& res) {
if (!cnt[x])
res++;
cnt[x]++;
}
void del(int x, int& res) {
cnt[x]--;
if (!cnt[x])
res--;
}
signed main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++)
scanf("%d", &w[i]);
for (int i = 0; i < m; i++) {
char op[2];
int a, b;
scanf("%s%d%d", op, &a, &b);
if (*op == 'Q') {
mq++;
q[mq] = {mq, a, b, mc};
} else {
c[++mc] = {a, b};
}
}
len = cbrt((double)n*mc)+1;
sort(q+1, q+mq+1, cmp);
int i = 0, j = 1, t = 0, res = 0;
for (int k = 1; k <= mq; k++) {
int id = q[k].id, tm = q[k].t;
int l = q[k].l, r = q[k].r;
while (i < r) add(w[++i], res);
while (i > r) del(w[i--], res);
while (j < l) del(w[j++], res);
while (j > l) add(w[--j], res);
while (t < tm) {
t++;
if (c[t].p>=j && c[t].p<=i) {
del(w[c[t].p], res);
add(c[t].c, res);
}
swap(w[c[t].p], c[t].c);
}
while (t > tm) {
if (c[t].p>=j && c[t].p<=i) {
del(w[c[t].p], res);
add(c[t].c, res);
}
swap(w[c[t].p], c[t].c);
t--;
}
ans[id] = res;
}
for (int i = 1; i <= mq; i++)
cout << ans[i] << "\n";
return 0;
}
