洛谷P2234 [HNOI2002]营业额统计（BZOJ1588）

平衡树

洛谷题目传送门

BZOJ题目传送门

改了三天TAT，ZZK说最后询问的时候也splay一下然后就过了。。。Orz ZZK

Splay第一题。。。然而发现Treap也能做。。。

每次插入和查询的时候乱Splay就好了，至于Splay和rotate怎么打。。。看代码吧

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 50000
#define inf 1000000000
using namespace std;
struct node{
    int w,fa,to[2];
}t[N+5];
int n,nd,size,rt,t1,t2,p1,p2,ans;
void rtt(int x,int &w){
    int y=t[x].fa,z=t[y].fa,l,r;
    if (t[y].to[0]==x) l=0; else l=1; r=l^1;
    if (y==w) w=x;
    else if (t[z].to[0]==y) t[z].to[0]=x;
    else t[z].to[1]=x;
    t[x].fa=z,t[y].fa=x,t[t[x].to[r]].fa=y;
    t[y].to[l]=t[x].to[r],t[x].to[r]=y;
}
void splay(int x,int &w){
    while (x!=w){
        int y=t[x].fa,z=t[y].fa;
        if (y!=w)
            if ((t[y].to[0]==x)^(t[z].to[0]==y)) rtt(x,w);
            else rtt(y,w);
        rtt(x,w);
    }
}
void nsrt(int &x,int w,int fa){
    if (!x){
        t[x=++nd].w=w,t[x].fa=fa;
        splay(x,rt); return;
    }
    nsrt(t[x].to[t[x].w<w],w,x);
}
void srch_frnt(int x,int w){
    if(!x) return;
    if(t[x].w<=w) p1=x,t1=t[p1].w,srch_frnt(t[x].to[1],w);
    else srch_frnt(t[x].to[0],w);
}
void srch_bck(int x,int w){
    if(!x) return;
    if(t[x].w>=w) p2=x,t2=t[p2].w,srch_bck(t[x].to[0],w);
    else srch_bck(t[x].to[1],w);
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<=n;i++){
        int x; scanf("%d",&x);
        p1=p2=rt;t1=-inf,t2=inf;
        srch_frnt(rt,x),srch_bck(rt,x);
        splay(p1,rt); splay(p2,rt);
        if (i==1) ans+=x; else ans+=min(x-t1,t2-x);
        nsrt(rt,x,0);
    }
    return printf("%d\n",ans),0;
}