求區間不同數的個數【樹狀陣列求解】

樹狀陣列求解

離線

模板

#include<iostream>
#include<algorithm>

using namespace std;
const int N=300005,M=200005;
struct Query{
	int l,r,id;
	bool operator<(const Query &a)const
	{
		return r==a.r?(l<a.l):(r<a.r);
	}
}q[M];

int n,sum[N],a[N],ans[N],vis[N];

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

void change(int x,int p)
{
	while(x<=n){
		sum[x]+=p;
		x+=lowbit(x);
	}
}

int getsum(int x)
{
	int ans=0;
	while(x>0){
		ans+=sum[x];
		x-=lowbit(x);
	}
	return ans;
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	  scanf("%d",&a[i]);
	int Q;
	scanf("%d",&Q);
	for(int i=0;i<Q;i++){
		scanf("%d%d",&q[i].l,&q[i].r);
		q[i].id=i;
	}
	sort(q,q+Q);
	int num=0;
	for(int i=1;i<=n&&num<Q;i++){
		if(vis[a[i]])
		  change(vis[a[i]],-1);
		change(i,1);
		vis[a[i]]=i;
		while(q[num].r<=i&&num<Q){
			ans[q[num].id]=getsum(q[num].r)-getsum(q[num].l-1);
			num++;
		}
	}
	for(int i=0;i<Q;i++)
	  printf("%d\n",ans[i]);
	return 0;
}