SPOJ D-query 主席树在线(求区间不同的数的个数)

点击打开链接

题意:n,m<=1e5,求[l,r]内不同的数的个数 

主席树:对于每个前缀i建立线段树,每个线段树中相同的数保留一次

此时l,r表示下标,sum:前缀i内:下标l~r不同的数的个数

 对于每个查询[l,r],只要在前缀r的线段树上,找到下标>=l的sum即可 

#include <bits/stdc++.h>
using namespace std;
const int N=2e6+20;
struct node{
int l,r,sum;
//sum:前缀i内:下标l~r不同的数的个数
}T[N*40];
int x,y,n,m,a
,last
;
int cnt,root
;
int build(int l,int r)
{
int rt=++cnt;
T[rt].sum=0,T[rt].l=T[rt].r=0;
if(l==r) return rt;
int m=(l+r)>>1;
T[rt].l=build(l,m);
T[rt].r=build(m+1,r);
return rt;
}
void update(int l,int r,int &x,int y,int v,int pos)
{
T[++cnt]=T[y],T[cnt].sum+=v,x=cnt;
if(l==r) return;
int m=(l+r)>>1;
if(m>=pos)
update(l,m,T[x].l,T[y].l,v,pos);
else
update(m+1,r,T[x].r,T[y].r,v,pos);
}
int query(int l,int r,int c,int pos)
{
if(l==r) return T[c].sum;
int m=(l+r)>>1;
if(m>=pos)
return T[T[c].r].sum+query(l,m,T[c].l,pos);
else
return query(m+1,r,T[c].r,pos);
}
int main()
{
while(cin>>n)
{
cnt=0;
memset(last,-1,sizeof(last));
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
root[0]=build(1,n);
for(int i=1;i<=n;i++)
{
int v=a[i];
if(last[v]==-1)
update(1,n,root[i],root[i-1],1,i);
else
{
update(1,n,root[i],root[i-1],-1,last[v]);
update(1,n,root[i],root[i],1,i);
}
last[v]=i;
}

cin>>m;
for(int i=0;i<m;i++)
{
scanf("%d%d",&x,&y);
int ans=query(1,n,root[y],x);
printf("%d\n",ans);
}
}
return 0;
}