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


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值