SPOJ DQUERY - D-query(主席树)

题目链接：https://www.spoj.com/problems/DQUERY/

 

题目大意：区间不同数个数

 

题目思路：T[i]这棵线段树表示的是1~i中数字存在情况，每个数字只保留最后一个位置，在这个最后一个位置上为1，即线段树里的值要么是0要么是1，是1的就是这个数字在1~i区间内最靠右的位置，所以当找在l~r中不同数个数的时候就是找T[r]线段树里l~n内数字的和

 

以下是代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
const int MAXN = 5e5+5;
const int MOD = 1e9+7;
int n,q,l,r,k,L[MAXN<<5],R[MAXN<<5],sum[MAXN<<5],T[MAXN],pos[1000005];
int x,tot;
int update(int pre,int l,int r,int x,int val){
    int rt=++tot;
    L[rt]=L[pre],R[rt]=R[pre],sum[rt]=sum[pre]+val;
    int mid=(l+r)>>1;
    if(l<r){
        if(x<=mid)L[rt]=update(L[pre],l,mid,x,val);
        else R[rt]=update(R[pre],mid+1,r,x,val);
    }
    return rt;
}
int query(int u,int l,int r,int x){
    if(l==r)return sum[u];
    int mid=(l+r)>>1;
    if(x<=mid)return query(L[u],l,mid,x)+sum[R[u]];
    else return query(R[u],mid+1,r,x);
}
int main(){
    while(~scanf("%d",&n)){
        tot=0;
        memset(T,0,sizeof(T));
        memset(L,0,sizeof(L));
        memset(R,0,sizeof(R));
        memset(sum,0,sizeof(sum));
        memset(pos,0,sizeof(pos));
        rep(i,1,n){
            scanf("%d",&x);
            if(pos[x]){
                int temp=update(T[i-1],1,n,pos[x],-1);
                T[i]=update(temp,1,n,i,1);
            }
            else{
                T[i]=update(T[i-1],1,n,i,1);
            }
            pos[x]=i;
        }
        scanf("%d",&q);
        while(q--){
            scanf("%d%d",&l,&r);
            printf("%d\n",query(T[r],1,n,l));
        }
    }
    return 0;
}