P1972 [SDOI2009] HH的项链（莫队算法）

题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。

有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答…… 因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

输入格式

一行一个正整数 n，表示项链长度。
第二行 n 个正整数 ai​，表示项链中第 i 个贝壳的种类。

第三行一个整数 m，表示 H 询问的个数。
接下来 m 行，每行两个整数 l,r，表示询问的区间。

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

2
2
4

思路：

莫队算法 = 离线 + 暴力 + 分块

基础莫队算法是一种离线算法。他通常用于不修改只查询的一类区间问题，复杂度为 O（nsqrt(n)）。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6;
struct node {
	int L, R, k; // k为查询操作的原始顺序
}q[N];
int pos[N];
int ans[N];
int cnt[N];
int a[N];
bool cmp(node a, node b) { // 莫队算法 
	if (pos[a.L] != pos[b.L]) { // 左端点排序 
		return pos[a.L] < pos[b.L]; // 块的大小排序
	}
	if (pos[a.L] & 1) return a.R > b.R; //奇偶性优化
	return a.R < b.R;
	
	if (a.L == b.L) return a.R < b.R;
	return a.L < b.L;
}
int ANS = 0;
void add(int x) {
	cnt[a[x]] ++;
	if (cnt[a[x]] == 1) ANS++;
}
void del(int x) {
	cnt[a[x]] --;
	if (cnt[a[x]] == 0) ANS--;
}
int main() {
	int n;
	scanf("%d", &n);
	int block = sqrt(n); // 每一块的大小
	for (int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
		pos[i] = (i - 1) / block + 1;//所属的块 从 1 ~ (n-1)/block + 1块
	}
	int m;
	scanf("%d", &m);
	for (int i = 1; i <= m; i++) {
		scanf("%d%d", &q[i].L, &q[i].R);
		q[i].k = i;
	}
	sort(q + 1, q + 1 + m, cmp);
	int L = 1, R = 0;
	for (int i = 1; i <= m; i++) {
		while (L < q[i].L) del(L++);
		while (R > q[i].R) del(R--);
		while (L > q[i].L) add(--L);
		while (R < q[i].R) add(++R);
		ans[q[i].k] = ANS;
	}
	for (int i = 1; i <= m; i++) {
		printf("%d\n", ans[i]);
	}
	return 0;
}