您的位置:首页 > 产品设计 > UI/UE

SPOJ DQUERY (主席树模板)

2016-08-08 15:32 281 查看
题意:

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

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

主席树大概是这么回事,每个结点记录前缀线段树,当然这里的线段树结点的申请是动态的,每次最多申请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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: