莫队——DQUERY - D-query

题目链接

题目大意：

给出一个长度为 n 的数列，a₁,a₂,…,a_n，有 q 个询问，每个询问给出数对 (i,j)，需要你给出 a_i,a_i+1,…,a_j这一段中有多少不同的数字。

输入输出样例

输入

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

输出

3
2
3

这题我是用莫队过的，莫队算法在询问排序后，通过移动左右端点来更改答案，从而保证它优秀的复杂度，具体可以看这篇博客（大佬太强了）。

#include<bits/stdc++.h>
using namespace std;
#define maxn 1000010
int n, m, L = 1, R, sum;
int a[maxn], f[maxn], ans[maxn], mp[maxn];

struct node{
	int l, r, id;
}q[maxn];

bool cmp(node x, node y){
	return f[x.l] ^ f[y.l] ? f[x.l] < f[y.l] : ((f[x.l] & 1) ? x.r < y.r : x.r > y.r);
}

void write(int x)
{
	if(x / 10){
		write(x / 10);
	}
	putchar(x % 10 + '0');
}

int main()
{
	scanf("%d", &n);	
	int unit = sqrt(n); 
	for(int i = 1; i <= n; i++){
		f[i] = i / unit + 1;
	}
	for(int i = 1; i <= n; i++){
		scanf("%d", &a[i]);
	}
	scanf("%d",&m);
	for(int i = 1; i <= m; i++){
		scanf("%d%d", &q[i].l, &q[i].r);
		q[i].id = i;
	}
	
	sort(q + 1, q + m + 1, cmp);
	for(int i = 1; i <= m; i++){
		while(L < q[i].l){
			sum -= !--mp[a[L++]];
		}
		while(L > q[i].l){
			sum += !mp[a[--L]]++;
		}
		while(R < q[i].r){
			sum += !mp[a[++R]]++;
		}
		while(R > q[i].r){
			sum -= !--mp[a[R--]];
		}
		ans[q[i].id] = sum;
	}
	for(int i = 1; i <= m; i++){
		write(ans[i]);
		printf("\n");
	}
	return 0;
}