您的位置:首页 > 其它

hdu5919 主席树求区间不同数

2017-03-09 22:15 369 查看
题目大意:有长度为n的序列,强制在线询问[l,r] 这段区间中所有不同数出现的第一个位置,按照位置从小到大排完序以后的中间(向上取整)的那个位置是多少?

解题思路:把n个元素倒着插进主席树中,把上次出现的同一个值的元素的贡献值-1,新的这个值的贡献值+1,这样就可以把这个元素最左侧的位置的贡献保留。查询区间[L,R]时只要查询第L个版本的主席树就好了。

ps:前一篇博客由于记录的是前缀,所以不能在主席树上进行类似二分的跳跃。
#include <bits/stdc++.h>
using namespace std;
#define maxn 210000
int treesiz,sum[maxn*40],num,pre[maxn];
int ls[maxn*40],rs[maxn*40],root[maxn],n,m,seq[maxn];
void update(int l,int r,int x,int &y,int pos,int f)
{
treesiz++;y=treesiz;sum[y]=sum[x]+f;
if(l==r)
{
return ;
}
ls[y]=ls[x];rs[y]=rs[x];
int mid=(l+r)>>1;
if(pos<=mid) update(l,mid,ls[x],ls[y],pos,f);
else update(mid+1,r,rs[x],rs[y],pos,f);
}
int query(int l,int r,int rt,int x,int y)
{
if(x<=l&&r<=y) return sum[rt];
int mid=(l+r)>>1;
int s1=0,s2=0;
if(x<=mid) s1=query(l,mid,ls[rt],x,y);
if(y>mid) s2=query(mid+1,r,rs[rt],x,y);
return s1+s2;
}
int solve(int l,int r,int rt,int num)
{
if(l==r) return l;
int mid=(l+r)>>1;
if(sum[ls[rt]]>=num) return solve(l,mid,ls[rt],num);
else return solve(mid+1,r,rs[rt],num-sum[ls[rt]]);
}
void cn()
{
memset(ls,0,sizeof(ls));
memset(rs,0,sizeof(rs));
memset(sum,0,sizeof(sum));
memset(pre,0,sizeof(pre));
memset(root,0,sizeof(root));
treesiz=num=0;
}

int main()
{
int cas,a,b,c;
cin>>cas;
for(int ii=1;ii<=cas;ii++)
{

scanf("%d%d",&n,&m);
cn();
for(int i=1;i<=n;i++)
{
scanf("%d",&seq[i]);
}
for(int i=n;i;i--)
{
if(pre[seq[i]])
{
update(1,n,root[i+1],root[i],pre[seq[i]],-1);
update(1,n,root[i],root[i],i,1);
}
else
{
update(1,n,root[i+1],root[i],i,1);
}
pre[seq[i]]=i;
}
int l,r,ans=0;
printf("Case #%d:",ii);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&l,&r);
l = (l+ans)%n+1;
r = (r+ans)%n+1;
if(l>r) swap(l,r);
int len = ( query(1,n,root[l],l,r)+1) / 2;
ans=solve(1,n,root[l],len);
printf(" %d",ans);
}
printf("\n");
}
return 0;
}
/*
2
5 2
3 3 1 5 4
2 2
4 4
5 2
2 5 2 1 2
2 3
2 4

1
5 2
2 5 2 1 2
2 3
2 4
*/
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: