#6085. 「美团 CodeM 资格赛」优惠券

题目描述

用last[x]表示对x进行的上一次操作的位置，vis[x]表示x是否在大楼内。

Splay维护'?'的位置。

若x要进楼：

1.若x已在楼内，则去找last[x]到i之间是否有'?'，若有，则可以把这个'?'当做是上一个x出楼，反之则 出现错误。

2.若x不在楼内，标记x在楼内。

若x要出楼：

1.若x在楼内，标记x不在楼内。

2.若x不在楼内，相同的去找last[x]到i之间是否有'?'。

#include<complex>
#include<cstdio>
using namespace std;
const int N=5e5+;
int n,sz,rot,ans=-;
int key[N],fat[N],son[N][],siz[N],last[N];
bool vis[N];
int qread()
{
    int x=;
    char ch=getchar();
    while(ch<'' || ch>'')
    {
        if(ch=='I')return ;
        if(ch=='O')return ;
        if(ch=='?')return ;
        ch=getchar();
    }
    while(ch>='' && ch<=''){x=x*+ch-'';ch=getchar();}
    return x;
}
void Update(int rt)
{
    siz[rt]=siz[son[rt][]]+siz[son[rt][]]+;
}
void Rotate(int x,int &rt)
{
    int a=fat[x],b=fat[a],l=son[a][]==x,r=l^;
    if(a==rt)rt=x;
    else son[b][son[b][]==a]=x;
    fat[son[x][r]]=a;fat[a]=x;fat[x]=b;
    son[a][l]=son[x][r];son[x][r]=a;
    Update(a);Update(x);
}
inline void Splay(int x,int &rt)
{
    while(x!=rt)
    {
        int a=fat[x],b=fat[a];
        if(a!=rt)
            if((son[a][]==x)^(son[b][]==a))
                Rotate(x,rt);
            else Rotate(a,rt);
        Rotate(x,rt);
    }
}
void Insert(int v,int k)
{
    int fa=;
    while(k && key[k]!=v)
        fa=k,k=son[k][key[k]<v];
    k=++sz;
    key[k]=v;siz[k]=;fat[k]=fa;
    if(fa)son[fa][key[fa]<v]=k;
    Splay(k,rot);
}
void Find(int x,int k)
{
    while(son[k][key[k]<x] && x!=key[k])
        k=son[k][key[k]<x];
    Splay(k,rot);
}
void Delete(int x)
{
    Find(x,rot);
    if(son[rot][] && son[rot][])
    {
        int tmp=rot,k=son[rot][];
        rot=k;
        while(son[k][])k=son[k][];
        siz[k]+=siz[son[tmp][]];
        fat[son[tmp][]]=k;son[k][]=son[tmp][];
        Splay(k,rot);
    }
    else rot=son[rot][]+son[rot][];
    fat[rot]=;
}
int GetNxt(int x)
{
    Find(x,rot);
    if(key[rot]>x)return rot;
    int k=son[rot][];
    while(son[k][])k=son[k][];
    return k;
}
int main()
{
    scanf("%d",&n);
    int tmp,p,x;
    for(int i=;i<=n;i++)
    {
        p=qread();
        if(p==)
        {
            x=qread();
            if(vis[x])
            {
                tmp=key[GetNxt(last[x])];
                if(!tmp)
                {
                    ans=i;
                    break;
                }
                else Delete(tmp);
            }
            vis[x]=;
            last[x]=i;
        }
        if(p==)
        {
            x=qread();
            if(!vis[x])
            {
                tmp=key[GetNxt(last[x])];
                if(!tmp)
                {
                    ans=i;
                    break;
                }
                else Delete(tmp);
            }
            vis[x]=;
            last[x]=i;
        }
        if(p==)Insert(i,rot);
    }
    printf("%d\n",ans);
    return ;
}