其他分享
首页 > 其他分享> > 买礼物(链表+线段树)

买礼物(链表+线段树)

作者:互联网

链接: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;
}

 

标签:柜子,val,线段,tree,链表,include,LL,礼物
来源: https://blog.csdn.net/zstuyyyyccccbbbb/article/details/113794700