【线段树+双向链表】2021牛客寒假算法基础集训营3 E-买礼物

题面

https://ac.nowcoder.com/acm/contest/9983/E

题意

两种操作

1. 把位置x上的数变成0
2. 查询【l，r】区间上有没有相同的两个数

思路

1. 题意转化：可以把问题转化成对于【l，r】这个区间中是否存在一个数，使得跟该数前面最近跟它相同的数位置在【l，r】这个区间中
2. 首先我们可以预处理出每个数前面跟它相等的最近的数的位置，然后用线段树存储我们预处理的数组，我们只需要判断跟他相同的最大的位置是否在【l，r】这段区间里，就知道是否存在了
3. 那对于第一个操作，我们要如何进行更新？采取双链表的形式，当x位置变成0以后，显然它就可以当作无用了，只需要更新前一个的nxt和后面一个的pre即可

代码

#include <iostream>
#include<cmath>
#include<vector>
#include<cstring>
#define ls k<<1
#define rs k<<1|1
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int maxn=1e6+100;
const int mod=1e9+7;
struct
{
    int l,r,maxx;
}node[maxn<<2];
int a[maxn],pre[maxn],nxt[maxn],tmp_pre[maxn];
void up(int k)
{
    node[k].maxx=max(node[ls].maxx,node[rs].maxx);
}
void build(int k,int l,int r)
{
    node[k].l=l;node[k].r=r;
    if(l==r)
    {
        node[k].maxx=pre[l];
        return ;
    }
    int mid=l+r>>1;
    build(ls,l,mid);
    build(rs,mid+1,r);
    up(k);
}
void change(int k,int x,int val)
{
    if(node[k].l==node[k].r&&x==node[k].l)
    {
        node[k].maxx=val;
        return ;
    }
    int mid=node[k].l+node[k].r>>1;
    if(x<=mid) change(ls,x,val);
    if(mid+1<=x) change(rs,x,val);
    up(k);
}
int query(int k,int l,int r)
{
    if(l<=node[k].l&&node[k].r<=r)
    {
        return node[k].maxx;
    }
    int mid=node[k].l+node[k].r>>1;
    int ans=0;
    if(l<=mid) ans=max(ans,query(ls,l,r));
    if(mid+1<=r) ans=max(ans,query(rs,l,r));
    return ans;
}
int main()
{
    int n,q,op,l,r,x;
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        //pre是要用的数组,双向链表
        pre[i]=tmp_pre[a[i]];
        nxt[tmp_pre[a[i]]]=i;
        tmp_pre[a[i]]=i;
    }
    build(1,1,n);
    while(q--)
    {
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d",&x);
            pre[nxt[x]]=pre[x];
            nxt[pre[x]]=nxt[x];
            change(1,x,0);
            change(1,nxt[x],pre[x]);
            pre[x]=nxt[x]=0;
        }
        else
        {
            scanf("%d%d",&l,&r);
            if(query(1,l,r)>=l) printf("1\n");
            else printf("0\n");
        }
    }
    return 0;
}