算法之 莫队 区间处理神器orz

Posted 2020-10-03

例题：luogu P1972 [SDOI2009]HH的项链

题目背景

无

题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答……因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

输入输出格式

输入格式：

 

第一行：一个整数N，表示项链的长度。

第二行：N 个整数，表示依次表示项链中贝壳的编号（编号为0 到1000000 之间的整数）。

第三行：一个整数M，表示HH 询问的个数。

接下来M 行：每行两个整数，L 和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

 

输出格式：

 

M 行，每行一个整数，依次表示询问对应的答案。

 

输入输出样例

输入样例#1：

6
1 2 3 4 3 5
3
1 2
3 5
2 6

输出样例#1：

2
2
4

说明

数据范围：

对于100%的数据，N <= 50000，M <= 200000。

题解

莫队真是好玩orz，存一下模板ovo

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <algorithm>
 4 #include <cmath>
 5 #define maxn 50010
 6 #define maxm 200005
 7 using namespace std;
 8 int a[maxn],cor[maxn],ans[maxn];
 9 struct Q{
10     int l,r,id,belong;
11 }q[maxm];
12 int n,m,block,cnt;
13 int comp(Q a,Q b) {
14     if(a.belong == b.belong) return a.r<b.r;
15     return a.belong<b.belong;
16 }
17 void del(int x) {
18     cor[a[x]]--;
19     if(cor[a[x]] == 0) cnt--;
20 }
21 void add(int x) {
22     cor[a[x]]++;
23     if(cor[a[x]] == 1) cnt++;
24 }
25 void work() {
26     int l=1,r=0;
27     for(int i=1;i<=m;i++) {
28         while(l<q[i].l) del(l),l++;
29         while(l>q[i].l) l--,add(l);
30         while(r<q[i].r) r++,add(r);
31         while(r>q[i].r) del(r),r--;
32         ans[q[i].id]=cnt;
33     }
34 }
35 int main() {
36     scanf("%d",&n);
37     block=sqrt(n);
38     for(int i=1;i<=n;i++) scanf("%d",&a[i]);
39     scanf("%d",&m);
40     for(int i=1;i<=m;i++) {
41         scanf("%d%d",&q[i].l,&q[i].r);
42         q[i].id=i;
43         q[i].belong=(q[i].l-1)/block+1;
44     }
45     sort(q+1,q+1+m,comp);
46     work();
47     for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
48     return 0;
49 }