BZOJ 1058 [ZJOI2007]报表统计 Splay

这个题真是不爽，人家set又短又快，我这个又长又慢，差十几毫秒tle。。。

我是一个splay维护相邻的最差小值，一个是维护全局最小差值（在插入的时候用前驱和后继更新）

 

 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cmath>

#define N 2222222
#define INF 1LL<<60

using namespace std;

long long msg,num[N],val[N],a[N],b[N];
int fa[N],son[N][2];
int tot,cnt,root1,root2;
int q[N],top;
int n,m;
int hs[N],lt[N],rt[N];

inline void prt(int x)
{
    if(!x) return;
    prt(son[x][0]);
    printf("%lld    ",val[x]);
    prt(son[x][1]);
}

inline void link(int x,int y,int c)
{
    fa[x]=y; son[y][c]=x;
}

inline void rotate(int x,int c)
{
    int y=fa[x];
    link(x,fa[y],son[fa[y]][1]==y);
    link(son[x][!c],y,c);
    link(y,x,!c);
}

inline void splay(int x,int g,int &rt)
{
    while(fa[x]!=g)
    {
        int y=fa[x];
        int cy=son[fa[y]][1]==y,cx=son[y][1]==x;
        if(fa[y]==g) rotate(x,cx);
        else
        {
            if(cx==cy) rotate(y,cy);
            else rotate(x,cx);
            rotate(x,cy);
        }
    }
    if(!g) rt=x;
}

inline int getnum()
{
    if(top) return q[top--];
    return ++cnt;
}

inline void newnode(int y,int &x,long long sp)
{
    x=getnum();
    fa[x]=y; val[x]=sp;
    son[x][0]=son[x][1]=0;
}

inline void init()
{
    newnode(cnt=0,root1,-INF);
    newnode(root1,son[root1][1],INF);
    newnode(0,root2,-INF);
    newnode(root2,son[root2][1],INF);
}

inline void build(int &u,int l,int r,int f)
{
    if(l>r) return;
    int mid=(l+r)>>1;
    newnode(f,u,a[mid]);
    build(son[u][0],l,mid-1,u);
    build(son[u][1],mid+1,r,u);
}

inline void insert(long long sp,int &rt)
{
    int x=rt;
    while(son[x][sp>val[x]]) x=son[x][sp>val[x]];
    newnode(x,son[x][sp>val[x]],sp);
    splay(son[x][sp>val[x]],0,rt);
}

inline int find(long long sp,int rt)
{
    int x=rt;
    while(x)
    {
        if(val[x]==sp) return x;
        x=son[x][sp>val[x]];
    }
}

inline int getmax(int x)
{
    while(son[x][1]) x=son[x][1];
    return x;
}

inline int getmin(int x)
{
    while(son[x][0]) x=son[x][0];
    return x;
}

inline void del(long long sp,int &rt)
{
    int z=find(sp,rt),x,y;
    splay(z,0,rt);
    x=getmax(son[z][0]);
    y=getmin(son[z][1]);
    splay(x,0,rt); splay(y,x,rt);
    q[++top]=son[y][0];
    fa[son[y][0]]=0;
    son[y][0]=0;
}

inline void go()
{
    msg=INF;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%lld",&b[i]);
    for(int i=1;i<n;i++) a[i]=abs(b[i+1]-b[i]);
    sort(a+1,a+n);
    init();
    build(son[son[root1][1]][0],1,n-1,son[root1][1]);
    for(int i=1;i<=n;i++) a[i]=b[i];
    sort(a+1,a+1+n);
    for(int i=1;i<n;i++) msg=min(msg,abs(a[i]-a[i+1]));
    build(son[son[root2][1]][0],1,n,son[root2][1]);
    for(int i=1;i<=n;i++)
    {
        num[i]=b[i]; hs[i]=i;
        lt[i]=i-1; rt[i-1]=i;
        lt[i+1]=i; rt[i]=i+1;
    }
    tot=n+1;
    char str[14]; int c; long long d;
    while(m--)
    {
        scanf("%s",str);
        if(str[0]=='I')
        {
            scanf("%d%lld",&c,&d);
            insert(d,root2);
            if(msg!=0)
            {
                int x=getmax(son[root2][0]),y=getmin(son[root2][1]);
                msg=min(msg,min(abs(val[x]-d),abs(val[y]-d)));
            }
            if(rt[hs[c]]!=n+1) del(abs(num[hs[c]]-num[rt[hs[c]]]),root1);
            ++tot; num[tot]=d;
            lt[tot]=hs[c]; lt[rt[hs[c]]]=tot;
            rt[tot]=rt[hs[c]]; rt[hs[c]]=tot;
            hs[c]=tot;
            if(lt[hs[c]]!=0) insert(abs(num[hs[c]]-num[lt[hs[c]]]),root1);
            if(rt[hs[c]]!=n+1) insert(abs(num[hs[c]]-num[rt[hs[c]]]),root1);
        }
        else if(str[4]=='G')
        {
            splay(1,0,root1);splay(2,1,root1);
            printf("%lld\n",val[getmin(son[2][0])]);
        }
        else printf("%lld\n",msg);
    }
}
    

int main()
{
    go();
    return 0;
}