主席树 询问区间内有多少个不同的数字

#include <bits/stdc++.h>
#define ll long long
using namespace std;


const ll inf=1e9+10;
const int maxn=1e5+20;
char s[maxn];
int ls[maxn*20],rs[maxn*20],sum[maxn*20],root[maxn];
int pre[30],tot;


void updata(int &rt,int pre,int val,int l,int r,int pos)
{
    rt=++tot;
    ls[rt]=ls[pre];
    rs[rt]=rs[pre];
    sum[rt]=sum[pre]+val;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(pos<=mid) updata(ls[rt],ls[pre],val,l,mid,pos);
    else updata(rs[rt],rs[pre],val,mid+1,r,pos);
}



int query(int al,int ar,int l,int r,int rt)
{
    if(al<=l&&ar>=r) return sum[rt];
    int ans=0;
    int mid=(l+r)>>1;
    if(al<=mid) ans+=query(al,ar,l,mid,ls[rt]);
    if(ar>mid) ans+=query(al,ar,mid+1,r,rs[rt]);
    return ans;
}


int main()
{
//    ios::sync_with_stdio(false);
//    cin.tie(0);
//    cout.tie(0);
    scanf("%s",s+1);
    int n=strlen(s+1);
    printf("%d\n",n);
    tot=0;
    for(int i=1;i<=n;i++){
        if(pre[s[i]-'a']==0) updata(root[i],root[i-1],1,1,n,i);
        else{
            updata(root[i],root[i-1],-1,1,n,pre[s[i]-'a']);     //先把之前那个位置的标记删掉，再在现在的位置上加上
            updata(root[i],root[i],1,1,n,i);   //注意这个地方，是在自己继承自己的基础上做修改
        }
        pre[s[i]-'a']=i;
    }
    int q;
    scanf("%d",&q);
    while(q--){
        int al,ar;
        scanf("%d%d",&al,&ar);
        int ans=query(al,ar,1,n,root[ar]);    //在第ar棵树上找位置为al及之后的个数
        printf("%d\n",ans);
    }

    return 0;
}