E.买礼物(线段树查询区间是否有重复的值)

买礼物

题目传送门:

买礼物
在这里插入图片描述

思路:

pre[i]表示第i个礼物的上一个相同礼物的位置
nex[i]示第i个礼物的下一个相同礼物的位置
我们要维护的就是区间 l ~ r 间nex[i]的最小值或者pre[i]的最大值。
我这里维护的是pre[i]的最大值,当删除位置i的物品时,把pre[i]置成0,但需要注意的是因为删除了当前位置的礼物,所以我们还需要修改下一个位置的相同礼物的上一个位置,及 pre[ nex [i] ] = pre[i]。查询时看区间最大值是否大于等于r即可。

AC Code

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
int a[N],b[N],c[N];
map<int,int>mp;
map<int,int>ans;
struct node
{
    
    
    int l,r;
    int maxn;
}tr[N*4];
void pushup(int k)
{
    
    
    tr[k].maxn=max(tr[k<<1].maxn,tr[k<<1|1].maxn);
}
void build(int k,int l,int r)
{
    
    
    tr[k].l=l,tr[k].r=r;
    if(l==r)
    {
    
    
        tr[k].maxn=b[r];
        return ;
    }
    int mid=(l+r)/2;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    pushup(k);
}
void modify(int k,int x,int y)
{
    
    
    if(tr[k].l==x&&tr[k].r==x)
    {
    
    
        tr[k].maxn=y;
        return ;
    }
    int mid=(tr[k].l+tr[k].r)/2;
    if(x<=mid) modify(k<<1,x,y);
    else modify(k<<1|1,x,y);
    pushup(k);
}
int query(int k,int l,int r)
{
    
    
    if(tr[k].l>=l&&tr[k].r<=r)
        return tr[k].maxn;
    int mid=(tr[k].l+tr[k].r)/2;
    int maxn=0;
    if(l<=mid) maxn=query(k<<1,l,r);
    if(r>mid) maxn=max(maxn,query(k<<1|1,l,r));
    return maxn;
}
int main()
{
    
    
    int n,q;
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)
    {
    
    
        scanf("%d",&a[i]);
        b[i]=mp[a[i]];
        mp[a[i]]=i;
    }
    for(int i=n;i>=1;i--)
    {
    
    
        if(ans[a[i]]==0)
            c[i]=n+1;
        else c[i]=ans[a[i]];
        ans[a[i]]=i;
    }
    build(1,1,n);
    while(q--)
    {
    
    
        int op;
        scanf("%d",&op);
        if(op==1)
        {
    
    
            int x;
            scanf("%d",&x);
            modify(1,x,0);
            if(c[x]!=n+1) modify(1,c[x],b[x]);
            c[b[x]]=c[x];
            b[c[x]]=b[x];
        }
        else
        {
    
    
            int x,y;
            scanf("%d%d",&x,&y);
            int k=query(1,x,y);
            if(k>=x) printf("1\n");
            else printf("0\n");
        }
    }
    system("pause");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Stevenwuxu/article/details/113764080
今日推荐