hdu5919 主席树求区间不同数

70 篇文章 0 订阅
4 篇文章 0 订阅

题目大意:有长度为 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
*/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值