SP3267 DQUERY – D-query

题目大意：

给出一个长度为n 的数列，/(a_{1},a_{2},…,a_{n}/)，有q 个询问，每个询问给出数对 /((i,j)/)，需要你给出 /(a_{i},a_{i+1} ，…，a_j/) 这一段中有多少不同的数字

分析：

考虑到树状数组对于处理前缀和问题很方便，自然，尝试用树状数组来进行处理。设 /(tree[i]/) 表示前 /(i/) 个数不同的数的种类数。显然，这样处理是不够的。在计算的过程中，我们也要对 /(tree[i]/) 进行更新。
为了更方便地进行 /(tree[i]/) 的更新，我们将每个询问先保存下来，进行离线操作，按照 /(j/) 的由小到大的顺序进行排序。对于每个区间，尽可能统计其靠近右端点的不同的数的个数，用一个 /(last[i]/) 表示在输入序列下标为 /(i/) 的元素左边第一个值与它相同的元素的下标。对于一个数，它最后出现的位置越靠近当前枚举的右端点越好。以 /(k/) 为枚举的下标，如果 /(a[k]/) 之前有相同的数，那么减掉之前那个数的贡献，加上当前数的贡献。这样就能完成对 /(tree/) 数组的维护。

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1e6 + 5;
int n,a[MAXN],tree[MAXN],rec[MAXN],last[MAXN],m,ans[MAXN];
struct askk{
	int l,r,pos;
}ask[MAXN]; 
bool cmp(askk aa,askk b){
	return aa.r < b.r;
}
int lowbit(int i){
	return i & -i;
}
void add(int x,int y){
	while(x <= n){
		tree[x] += y;
		x += lowbit(x);
	}
}
int sum(int x){
	int ans = 0;
	while(x > 0){
		ans += tree[x];
		x -= lowbit(x);
	}
	return ans;
}

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin >> n;
	for(int i = 1; i <= n; i++){
		cin >> a[i];
		last[i] = rec[a[i]]?rec[a[i]]:MAXN;
		rec[a[i]] = i;//维护 last 数组
	}
	cin >> m;
	for(int i = 1; i <= m; i++){
		cin >> ask[i].l >> ask[i].r;
		ask[i].pos = i;
	}
	int k = 1;
	sort(ask + 1,ask +1 + m,cmp);//对询问排序
	for(int i = 1; i <= m; i++){
		while(k <= ask[i].r){
			add(last[k],-1);
			add(k,1);//维护tree数组
			k++;
		}
		ans[ask[i].pos] = sum(ask[i].r) - sum(ask[i].l - 1);
	}
	for(int i = 1; i <= m; i++){
		cout << ans[i] << "/n";
	}
}