1. 程式人生 > >求區間不同數的個數【主席樹求解】

求區間不同數的個數【主席樹求解】

線上

模板

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;
const int N=100010;
int a[N],b[N],rt[N*20],ls[N*20],rs[N*20],sum[N*20],vis[N*20];

int id;
void build(int &o,int l,int r)
{
	o=++id;
	sum[o]=0;
	if(l==r) return;
	int m=(l+r)>>1;
	build(ls[o],l,m);
	build(rs[o],m+1,r);
}

void update(int &o,int l,int r,int last,int pos,int val)
{
	o=++id;
	ls[o]=ls[last];
	rs[o]=rs[last];
	sum[o]=sum[last]+val;
	if(l==r) return;
	int m=(l+r)>>1;
	if(pos<=m)
	  update(ls[o],l,m,ls[last],pos,val);
	else
	  update(rs[o],m+1,r,rs[last],pos,val);
}
//查詢區間[L,R]內小於等於k的個數 
int query(int pos,int root,int l,int r)
{
	if(l==r)
	  return sum[root];
	int m=(l+r)>>1;
	if(m>=pos)
	  return sum[rs[root]]+query(pos,ls[root],l,m);
	else
	  return query(pos,rs[root],m+1,r);
}

int main()
{
	int T,n,q,l,r,k;
	scanf("%d",&T);
	for(int t=1;t<=T;t++){
		memset(vis,0,sizeof(vis));
		scanf("%d%d",&n,&q);
		for(int i=1;i<=n;i++)
		  scanf("%d",&a[i]);
		id=0;
		build(rt[0],1,n);
		for(int i=1;i<=n;i++){
			if(!vis[a[i]])//如果a[i]還未出現過 
			  update(rt[i],1,n,rt[i-1],i,1);
			else{//如果a[i]已經出現了 
				int pos=vis[a[i]];
				int temp;
				update(temp,1,n,rt[i-1],pos,-1);
				update(rt[i],1,n,temp,i,1);
			}
			vis[a[i]]=i;
		}
		while(q--){
			scanf("%d%d",&l,&r);
			int num=query(l,rt[r],1,n);
			printf("%d\n",num);
		}
	}
	return 0;
}