bzoj1588: [HNOI2002]营业额统计 SBT&&Splay

所有操作都有，就当练手吧。。。貌似数据有问题，要特判一下。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <queue>
#include <cmath>
using namespace std;
#define maxn 1000000
int root,ls[maxn],rs[maxn],siz[maxn],val[maxn];
int n,cnt,m;
bool vis[2000000];
inline void rotate_l(int& x)
{
    int y=rs[x];
    rs[x]=ls[y];
    ls[y]=x;
    siz[y]=siz[x];
    siz[x]=siz[ls[x]]+siz[rs[x]]+1;
    x=y;
}
inline void rotate_r(int& x)
{
    int y=ls[x];
    ls[x]=rs[y];
    rs[y]=x;
    siz[y]=siz[x];
    siz[x]=siz[ls[x]]+siz[rs[x]]+1;
    x=y;
}
inline void maintain(int& x,int fg)
{
    if(!fg)
    {
        if(siz[ls[ls[x]]]>siz[rs[x]]) rotate_r(x);
        else if(siz[rs[ls[x]]]>siz[rs[x]]) rotate_l(ls[x]),rotate_r(x);
        else return;
    }
    else
    {
        if(siz[rs[rs[x]]]>siz[ls[x]]) rotate_l(x);
        else if(siz[ls[rs[x]]]>siz[ls[x]]) rotate_r(rs[x]),rotate_l(x);
        else return;
    }
    maintain(ls[x],0);maintain(rs[x],1);
    maintain(x,0);maintain(x,1);
}
inline void insert(int& x,int b)
{
    if(x==0)
    {
        cnt++;
        x=cnt;
        ls[x]=rs[x]=0;
        siz[x]=1;
        val[x]=b;
    }
    else
    {
        siz[x]++;
        if(b>val[x])
        {
            insert(rs[x],b);
        }
        else
        {
            insert(ls[x],b);
        }
        maintain(x,b>=val[x]);
    }
}
inline int select(int x,int rk)
{
    if(siz[ls[x]]+1==rk)
    {
        return val[x];
    }
    else if(siz[ls[x]]+1>rk) return select(ls[x],rk);
    return select(rs[x],rk-siz[ls[x]]-1);
}
 
int pred(int& x,int y,int b)
{
    if(!x) return y;
    if(val[x]<b) return pred(rs[x],x,b);
    else return pred(ls[x],y,b);
}
int succ(int& x,int y,int b)
{
    if(!x) return y;
    if(val[x]>b) return succ(ls[x],x,b);
    else return succ(rs[x],y,b);
}
int main()
{
    int a,b,f1,f2;
    char str[10];
    scanf("%d",&n);
    int ans=0;
    scanf("%d",&ans);
    insert(root,ans);
    vis[ans+maxn]=1;
 
    for(int i=2;i<=n;i++)
    {
 
        if(scanf("%d",&b)==EOF) b=0;
        if(vis[b+maxn]) continue;
        vis[b+maxn]=1;
 
        f1=pred(root,0,b);
        f2=succ(root,0,b);
 
        if(f1!=0) f1=val[f1];
        else f1=-maxn;
        if(f2!=0) f2=val[f2];
        else f2=-maxn;
        f1=abs(f1-b);
        f2=abs(f2-b);
 
        ans+=min(f1,f2);
        insert(root,b);
    }
 
    printf("%d\n",ans);
    return 0;
}

splay版 生平第一个splay，比SBT慢好多。。。。不过运用很灵活。

#include<iostream>
#include<cstdio>
#define inf 1000000000
using namespace std;
int ans,n,t1,t2,root,cnt;
int tr[51000][2],fa[51000],siz[51000],val[51000];
void rotate(int x,int& rt)
{
    int y=fa[x],z=fa[y],l,r;
    if(tr[y][0]==x) l=0;
    else l=1;
    r=l^1;
    if(y==rt) rt=x;
    else
    {
        if(tr[z][0]==y) tr[z][0]=x;
        else tr[z][1]=x;
    }
    fa[x]=z;
    fa[y]=x;
    fa[tr[x][r]]=y;
    tr[y][l]=tr[x][r];
    tr[x][r]=y;
}
void splay(int x,int& rt)
{
    int y,z;
	while(x!=rt)
	{
		y=fa[x];z=fa[y];
        if(y!=rt)
        {
            if((tr[y][0]==x)^(tr[z][0]==y)) rotate(x,rt);
            else
            {
                rotate(y,rt);
            }
        }
        rotate(x,rt);
	}
}
void pred(int x,int b)
{
     if(x==0)return;
     if(val[x]<=b){t1=val[x],pred(tr[x][1],b);}
     else pred(tr[x][0],b);
}
void succ(int x,int b)
{
     if(x==0)return;
     if(val[x]>=b){t2=val[x],succ(tr[x][0],b);}
     else succ(tr[x][1],b);
}
void insert(int &x,int last,int b)
{
    if(!x)
    {
        cnt++;
        x=cnt;
        tr[x][0]=tr[x][1]=0;
        fa[x]=last;
        val[x]=b;
        splay(x,root);
        return ;
    }
    if(b<val[x]) insert(tr[x][0],x,b);
    else insert(tr[x][1],x,b);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
        int x;if(scanf("%d",&x)==EOF)x=0;
        t1=-inf;t2=inf;
	    pred(root,x);
	    succ(root,x);
	    if(i!=1)ans+=min(x-t1,t2-x);
	    else ans+=x;
	    insert(root,0,x);
	}
	printf("%d",ans);
	return 0;
}