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;}