hdu5919 主席树求区间不同数

来源:互联网 发布:小米6陶瓷 知乎 编辑:程序博客网 时间:2024/05/29 07:57

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


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

ps:前一篇博客由于记录的是前缀,所以不能在主席树上进行类似二分的跳跃。

#include <bits/stdc++.h>using namespace std;#define maxn 210000int 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;}/*25 23 3 1 5 42 24 45 22 5 2 1 22 32 415 22 5 2 1 22 32 4*/

0 0