维护集合Ⅱ【Splay】

题目描述

如未特别说明，以下所有数据均为整数。

维护一个多重集 S ，初始为空，有以下几种操作：

1. 插入元素 x 。
2. 删除元素 x ，保证删除的元素一定存在。
3. 输出 S 中相差最小的两个元素的差，保证此时|S|≥2 。
   操作共 n 次 。

输入格式

第一行一个整数 n (1≤n≤3×10^5)，表示共有 n 次操作。
接下来 n 行，每行为以下几种格式之一：
0 x ，把 x（0≤x≤10^9 ）加入 S 。
1 x ，删除 x 。
2 ，回答询问：S 中相差最小的两个元素的差是多少？

输出格式

对于每次询问，输出单独一行表示答案。

样例数据 1

输入

5
0 3
0 7
2
0 4
2

输出

4
1

解题思路：

用Splay维护每个数与它的前驱（小于等于它的最大数）的差及子树中差的最小值即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#define ll long long
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c<'0'||c>'9')&&c!='-';c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=300005,INF=0x3f3f3f3f;
int tot,root,val[N],d[N],mi[N],fa[N],son[N][2];

int which(int x)
{
    return son[fa[x]][1]==x;
}

void Rotate(int x)
{
    int y=fa[x],z=fa[y],t=which(x);
    if(y!=root)son[z][which(y)]=x;
    else root=x;
    fa[x]=z,fa[y]=x;
    son[y][t]=son[x][t^1],son[x][t^1]=y;
    if(son[y][t])fa[son[y][t]]=y;
    mi[x]=mi[y];
    mi[y]=min(d[y],min(mi[son[y][0]],mi[son[y][1]]));
}

void Splay(int x,int f)
{
    while(fa[x]!=f)
    {
        if(fa[fa[x]]!=f)
            which(x)==which(fa[x])?Rotate(fa[x]):Rotate(x);
        Rotate(x);
    }
}

void Insert(int &u,int v,int f)
{
    if(!u)
    {
        u=++tot,val[u]=v,fa[u]=f;
        return;
    }
    if(v<=val[u])Insert(son[u][0],v,u);
    else Insert(son[u][1],v,u);
}

int getpre(int x)
{
    x=son[x][0];
    while(son[x][1])x=son[x][1];
    return x;
}

int getsuf(int x)
{
    x=son[x][1];
    while(son[x][0])x=son[x][0];
    return x;
}

void modify(int x)
{
    int y;
    Splay(x,0);
    if(son[x][0])
    {
        y=getpre(x);
        d[x]=val[x]-val[y];
        mi[x]=min(d[x],mi[x]);
    }
    if(son[x][1])
    {
        y=getsuf(x);
        Splay(y,x);
        d[y]=val[y]-val[x];
        mi[y]=min(d[y],min(mi[son[y][0]],mi[son[y][1]]));
        mi[x]=min(mi[y],mi[x]);
    }
}

int find(int u,int v)
{
    if(val[u]==v)return u;
    if(v<val[u])return find(son[u][0],v);
    return find(son[u][1],v);
}

void Delete(int v)
{
    int x=find(root,v);
    Splay(x,0);
    if(!son[x][0]&&!son[x][1])
    {
        root=0;
        return;
    }
    if(!son[x][0])
    {
        int y=getsuf(x);
        Splay(y,x);
        root=y,fa[y]=0;
        d[y]=INF,mi[y]=mi[son[y][1]];
        return;
    }
    if(!son[x][1])
    {
        root=son[x][0],fa[root]=0;
        return;
    }
    int y=getsuf(x),z=getpre(x);
    Splay(y,x),Splay(z,x);
    d[y]=val[y]-val[z];
    mi[y]=min(d[y],min(mi[z],mi[son[y][1]]));
    root=y,fa[y]=0,son[y][0]=z;
    fa[z]=y;
}

int main()
{
    //freopen("lx.in","r",stdin);
    //freopen("lx.out","w",stdout);
    memset(mi,INF,sizeof(mi));
    memset(d,INF,sizeof(d));
    int n,op,x;
    n=getint();
    while(n--)
    {
        op=getint();
        if(op==0)
        {
            x=getint();
            Insert(root,x,0);
            modify(tot);
        }
        else if(op==1)
        {
            x=getint();
            Delete(x);
        }
        else cout<<mi[root]<<'\n';
    }
    return 0;
}