SPOJ 3267 - DQUERY D-query [主席树]

题意：给出长度为n的数列，m个询问，每次询问区间【L，R】中不同数的个数

题解：这题与以往的主席树不同，用位置建树，第i个线段树维护前i个位置的信息每次向后移动位置分为两种情况

1、该位置的数之前没有出现过，则在第i个线段树的i这个位置加1

2、该位置上的数出现过，则在第i个线段树上，前一次出现的位置处（设为k）

AC代码：

#include<stdio.h>  
#include<map>  
using namespace std;  
#define N 30005  
int a[N],tot;  
int tree[N*25],lchild[N*25],rchild[N*25],root[N*25];  
map<int,int>mp;  
void update(int last,int cur,int L,int R,int x,int k)  
{  
    tree[cur]=tree[last];  
    lchild[cur]=lchild[last];  
    rchild[cur]=rchild[last];  
    if(L==R)  
    {  
        tree[cur]+=k;  
        return ;  
    }  
    int mid=L+R>>1;  
    if(x<=mid)update(lchild[last],lchild[cur]=++tot,L,mid,x,k);  
    else update(rchild[last],rchild[cur]=++tot,mid+1,R,x,k);  
    tree[cur]=tree[lchild[cur]]+tree[rchild[cur]];  
}  
int query(int root,int l,int r,int L,int R)  
{  
    if(l<=L&&R<=r)  
        return tree[root];  
    int mid=L+R>>1;  
    if(r<=mid)return query(lchild[root],l,r,L,mid);  
    else if(mid<l)return query(rchild[root],l,r,mid+1,R);  
    else return query(lchild[root],l,mid,L,mid)+query(rchild[root],mid+1,r,mid+1,R);  
}  
int main()  
{  
    int n;  
    scanf("%d",&n);  
    for(int i=1;i<=n;i++)  
        scanf("%d",&a[i]);  
    for(int i=1;i<=n;i++)  
    {  
        if(!mp[a[i]]) update(root[i-1],root[i]=++tot,1,n,i,1);  
        else   
        {  
            update(root[i-1],root[i]=++tot,1,n,i,1);  
            int now=++tot;
            update(root[i],now,1,n,mp[a[i]],-1);  
            root[i]=now;
        }  
        mp[a[i]]=i;  
    }  
    int q;  
    scanf("%d",&q);  
    while(q--)  
    {  
        int l,r;  
        scanf("%d%d",&l,&r);  
        printf("%d\n",query(root[r],l,r,1,n));  
    }  
}