bzoj1588营业额统计

treap一发,每次寻找最接近的两个以前的点计算即可,听同学吐槽我的代码风格很像黄学长,恩没错我是学习他的代码风格的
#include<cstdio>
#include<ctime>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#define N 50005
#define inf 10000000
using namespace std;
int ls[N],rs[N],ra[N],key[N],siz[N],root,len;
int n,ans,t1,t2;

void push(int k)
{
        siz[k]=siz[ls[k]]+siz[rs[k]]+1;
}

void rturn(int &k)
{
        int t=ls[k];
        ls[k]=rs[t];
        rs[t]=k;
        siz[t]=siz[k];
        push(k);
        k=t;
}

void lturn(int &k)
{
        int t=rs[k];
        rs[k]=ls[t];
        ls[t]=k;
        siz[t]=siz[k];
        push(k);
        k=t;
}

void ins(int &k,int val)
{
        if (!k)
        {
                int t=++len;
                key[t]=val;
                siz[t]=1;
                ra[t]=rand();
                k=t;
                return;
        }
        siz[k]++;
        if (val>=key[k])
        {
                ins(rs[k],val);
                if (ra[rs[k]]<ra[k]) lturn(k);
        }
        else
        {
                ins(ls[k],val);
                if (ra[ls[k]]<ra[k]) rturn(k);
        }
}

void askbefore(int k,int val)
{
        if (!k) return;
        if (key[k]<=val)
        {
                t1=key[k];
                askbefore(rs[k],val);
        }
        else askbefore(ls[k],val);
}

void askafter(int k,int val)
{
        if (!k) return;
        if (key[k]>=val)
        {
                t2=key[k];
                askafter(ls[k],val);
        }
        else askafter(rs[k],val);
}

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;
                askbefore(root,x);
                askafter(root,x);
                if (i!=1) ans+=min(x-t1,t2-x);
                        else ans+=x;
                ins(root,x);
//                printf("%d %d %d %d\n",t1,t2,x,ans);
        }
        printf("%d",ans);
        return 0;
}

你可能感兴趣的:(数据结构,treap,平衡树,树堆)