SPOJ DQUERY (主席树模板)

题意:

给出一个序列,询问区间内有多少个不同的数


这题卡分块莫队,写了一下主席树,已加入模板

主席树大概是这么回事,每个结点记录前缀线段树,当然这里的线段树结点的申请是动态的,每次最多申请logn个,对于询问来说就只需要询问前缀r线段树中l到n区间内不同数的个数了


#include
#include
#include
#include
#include

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


你可能感兴趣的:(===数据结构===,主席树)