1. 程式人生 > >DQUERY D-query [主席樹]

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));  
    }  
}