P1972 [SDOI2009]HH的项链

传送门

思路

题意概括：问 [L,R] 里，有多少种数字 ？

在以 R 为右端点的区间里，对于某一种数字，我们只关心 它最后一次出现的位置
我们只在其最后一次 出现的位置上 加 1 ( add 操作 )，其余出现的位置都保持为 0

比如数列：1、3、3、1
当 R = 4 时，进行 add 操作后，各位置上的数为 0、0、1、1
此时，无论 左端点 L (L <= R)取值为多少，[L,R] 里的数字种类 都等于 sum（R）- sum（L - 1）

对于 m 个询问，若每次都遍历询问区间 [Li,Ri]，显然会超时
但我们可以将其转为离线做法，即因为上面的性质，也为了避免超时
==> 将 m 个询问的区间，按照 R 从小到大排序，然后依次处理

具体见代码

Code

#include <bits/stdc++.h>

using namespace std;

const int N = 2e6+10;

int a[N],tr[N];
int last[N],ans[N]; // last[i] ：数字 i 上一次出现的位置
int n,m;
struct node{
	int id;
	int l,r;
	bool operator < (const node & x) const{
		return r < x.r;
	}	
}p[N];

int lowbit(int x){
	return x & - x;
}

void add(int x,int v){
	for(int i=x;i<=N;i+=lowbit(i)) tr[i]+=v;
}

int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i)) res+=tr[i];
	return res;
}

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	
	scanf("%d",&m);
	for(int i=1;i<=m;i++){
		int l,r;
		scanf("%d%d",&l,&r);
		p[i]={i,l,r};
	}
	
	sort(p+1,p+1+m);
	
	// 双指针
	for(int i=1,j=1;i<=m;i++){
		int id=p[i].id,l=p[i].l,r=p[i].r;
		while(j<=r){
			if(last[a[j]]) add(last[a[j]],-1);  // 保证最后一个位置外的位置，上面的数字都为0
			add(j,1);  // j 为此时的 a[j] 出现的最后一个位置
			last[a[j]]=j;   // 更新 last
			j++;
		}
		ans[id]=sum(r)-sum(l-1);
	}
	
	for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
	
	return 0;
}