主席树练习3——SP3267 DQUERY - D-query

蒟蒻的垂死挣扎

求蛇皮的区间不重复的数个数。每次记录每个值最新的出现位置，查询时即查询右端点那棵树，这样可以保证每种数都被包括到且不重复，在外面开一个vis记录上一次出现位置即可实现。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#define RG register
#define N 200100
#define ll long long
#define ld long double
using namespace std;

inline ll read(){
  RG ll x=0,o=1; RG char ch=getchar();
  while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
  if(ch=='-') o=-1,ch=getchar();
  while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
  return x*o;
}

int len,n,m,a[N],b[N],yyb[N],tot,root[N];
struct mona { int le,ri; int w; } tr[N*80];

inline void Orz(){
    sort(yyb+1,yyb+1+n);
    len=unique(yyb+1,yyb+1+n)-yyb-1;
    for(RG int i=1;i<=n;++i) a[i]=lower_bound(yyb+1,yyb+1+len,a[i])-yyb;
}

inline void Build(RG int l,RG int r,RG int &x,RG int p,RG int w){
    tr[++tot]=tr[x],tr[x=tot].w+=w;
    if(l==r) return ;
    RG int mid=(l+r)>>1;
    if(p<=mid) Build(l,mid,tr[x].le,p,w);
    else Build(mid+1,r,tr[x].ri,p,w);
}
inline int Query(RG int l,RG int r,RG int y,RG int nl,RG int nr){
    if(!y) return 0;
    if(l==nl&&r==nr) return tr[y].w;
    RG int mid=(l+r)>>1;
    if(nr<=mid) return Query(l,mid,tr[y].le,nl,nr);
    else if(nl>mid) return Query(mid+1,r,tr[y].ri,nl,nr);
    else return Query(l,mid,tr[y].le,nl,mid)+Query(mid+1,r,tr[y].ri,mid+1,nr);
}

int main(){
    n=read();
    for(RG int i=1;i<=n;++i) a[i]=yyb[i]=read();
    Orz(),m=read();
    for(RG int i=1;i<=n;++i){
        root[i]=root[i-1];
        if(!b[a[i]]) { b[a[i]]=i,Build(1,n,root[i],i,1); continue ; }
        Build(1,n,root[i],b[a[i]],-1);
        b[a[i]]=i,Build(1,n,root[i],i,1);
    }
    for(RG int i=1;i<=m;++i){
        RG int x=read(),y=read();
        printf("%d\n",Query(1,n,root[y],x,y));
    }
}