买礼物(链表+线段树)

链接:https://ac.nowcoder.com/acm/contest/9983/E
来源:牛客网
 

在卖礼物的超市中有n个柜子,每个柜子里都摆放了一个礼物,每个礼物有自己的一个编号,第i个柜子里的礼物编号为ai​。
茶山牛想给牛牛和牛妹买相同编号的礼物,但礼物有可能在某个时刻被其他人买走,而且柜子数量太多,因此茶山牛在某个时刻只想知道某一个柜子区间是否能买到两件相同编号的礼物。
具体来说,有q次操作,格式如下:
1x,第x个柜子里的礼物被买走,保证此时这个柜子里的礼物还在。
2 l r,茶山牛询问第l到第r个柜子未被买走的礼物中是否有两个礼物编号相同。

输出描述:

对每次茶山牛的询问输出一行一个整数,如果在指定的区间内有两个礼物编号相同则输出1,否则输出0。

思路:预处理每个数的最近前后位置,类似链表,然后线段树存每个下标的next位置,查询看[l,r]的最小值(对应最近的nxet[])是否<=r。线段树上修改即修改last[i]的next[i]的位置。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=5e5+100;
typedef long long LL;
const LL INF=1e10;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL lst[maxn],nxt[maxn],a[maxn];
vector<LL>v[2*maxn];
struct Tree{
    LL l,r,val;
}tree[maxn*4];
void push_up(LL p){
    tree[p].val=min(tree[p*2].val,tree[p*2+1].val);
}
void build(LL p,LL l,LL r){
    tree[p].l=l;tree[p].r=r;tree[p].val=INF;
    if(l==r){tree[p].val=nxt[l];return;}
    LL mid=(l+r)>>1;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    push_up(p);
}
void modify(LL p,LL l,LL r,LL d){
    if(l<=tree[p].l&&r>=tree[p].r){
        tree[p].val=d;
        return;
    }
    LL mid=(tree[p].l+tree[p].r)>>1;
    if(l<=mid) modify(p*2,l,r,d);
    if(r>mid) modify(p*2+1,l,r,d);
    push_up(p);
}
LL query(LL p,LL l,LL r){
    if(l<=tree[p].l&&r>=tree[p].r){
        return tree[p].val;
    }
    LL ans=INF;
    LL mid=(tree[p].l+tree[p].r)>>1;
    if(l<=mid) ans=min(ans,query(p*2,l,r));
    if(r>mid)  ans=min(ans,query(p*2+1,l,r));
    return ans;
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  LL n,q;cin>>n>>q;

  for(LL i=1;i<=n;i++) cin>>a[i];
  for(LL i=1;i<=1e6;i++) v[i].push_back(0);
  for(LL i=1;i<=n;i++){
    v[a[i]].push_back(i);
  }
  for(LL i=1;i<=1e6;i++) v[i].push_back(INF);

  for(LL i=1;i<=1e6;i++){
    for(LL j=1;j<v[i].size()-1;j++){
        lst[v[i][j]]=v[i][j-1];
        nxt[v[i][j]]=v[i][j+1];
    }
  }
  build(1,1,n);
  while(q--){
    LL x,l,r;cin>>x;
    if(x==1){
        cin>>l;
        modify(1,l,l,INF);
        modify(1,lst[l],lst[l],nxt[l]);
        nxt[lst[l]]=nxt[l];

        if(nxt[l]<INF){
            lst[nxt[l]]=lst[l];
        }
        lst[l]=0;
        nxt[l]=INF;
    }
    else if(x==2){
        cin>>l>>r;
        LL t=query(1,l,r);
        if(t<=r){
            cout<<"1"<<"\n";
        }
        else cout<<"0"<<"\n";
    }
  }
return 0;
}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值