1. 程式人生 > >SPOJ DQUERY (主席樹模板)

SPOJ DQUERY (主席樹模板)

題意:

給出一個序列,詢問區間內有多少個不同的數

這題卡分塊莫隊,寫了一下主席樹,已加入模板

主席樹大概是這麼回事,每個結點記錄字首線段樹,當然這裡的線段樹結點的申請是動態的,每次最多申請logn個,對於詢問來說就只需要詢問字首r線段樹中l到n區間內不同數的個數了

#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<map>

using namespace std;

#define maxn 30004

int a[maxn],root[maxn],tot;

struct node
{
    int l,r;
    int s;
    node()
    {
        l=r=s=0;
    }
} t[30*maxn];

int newnode(int s,int l,int r)
{
    int rt=++tot;
    t[rt].s=s;
    t[rt].l=l;
    t[rt].r=r;
    return rt;
}

void update(int &rt,int pre,int pos,int l,int r,int val)
{
    node &temp=t[pre];
    rt=newnode(temp.s+val,temp.l,temp.r);
    if(l==r) return ;
    int mid=l+r>>1;
    if(pos<=mid) update(t[rt].l,temp.l,pos,l,mid,val);
    else update(t[rt].r,temp.r,pos,mid+1,r,val);
}

int query(int rt,int pos,int l,int r)
{
    if(l==pos) return t[rt].s;
    int mid=l+r>>1;
    if(pos<=mid) return query(t[rt].l,pos,l,mid)+t[t[rt].r].s;
    return query(t[rt].r,pos,mid+1,r);
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        tot=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
        }
        map<int ,int >mp;
        int temp;
        for(int i=1; i<=n; i++)
        {
            int e=a[i];
            if(!mp[e])
            {
                update(root[i],root[i-1],i,1,n,1);
            }
            else
            {
                update(temp,root[i-1],mp[e],1,n,-1);
                update(root[i],temp,i,1,n,1);
            }
            mp[e]=i;
        }
        int q;
        scanf("%d",&q);
        while(q--)
        {
            int l,r;
            scanf("%d%d",&l,&r);
            printf("%d\n",query(root[r],l,1,n));
        }
    }
    return 0;
}