bzoj 1878: [SDOI2009]HH的项链(莫队算法)

Description

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

Input

第一行：一个整数N，表示项链的长度。 第二行：N个整数，表示依次表示项链中贝壳的编号（编号为0到1000000之间的整数）。 第三行：一个整数M，表示HH询问的个数。 接下来M行：每行两个整数，L和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

Output

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

Sample Input

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

Sample Output

2
2

4

solution：

莫队算法模板题，只需要自己修改O（1）的部分。

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn = 500050;
int unit;
int num[1000050], a[maxn];
int tmp, ans[maxn];
struct Query
{
	int i, l, r;
	bool operator<(const Query &q)const
	{
		if (l / unit == q.l / unit) return r < q.r;
		return l / unit < q.l / unit;
	}
}query[maxn];
void insert(long long x)
{
	if (num[x] == 0)tmp++;
	++num[x];
}
void remove(long long x)
{
	if (num[x] == 1)tmp--;
	--num[x];
}
int main(){
	int n, t;
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
	unit = (int)sqrt((double)n);
	scanf("%d", &t);
	for (int i = 0; i < t; ++i)
	{
		query[i].i = i;
		scanf("%d%d", &query[i].l, &query[i].r);
	}
	sort(query, query + t);
	int l = 1, r = 0; 
	tmp = 0;
	for (int i = 0; i<t; ++i)
	{
		while (l < query[i].l) remove(a[l++]);
		while (l > query[i].l) insert(a[--l]);
		while (r>query[i].r) remove(a[r--]);
		while (r < query[i].r) insert(a[++r]);
		ans[query[i].i] = tmp;
	}
	for (int i = 0; i < t; ++i)
		printf("%d\n", ans[i]);
	return 0;
}