HNOI2002 营业额统计（SBT模板题&splay解法）

1588: [HNOI2002]营业额统计

时间限制: 5 Sec   内存限制: 162 MB
提交: 1897   解决: 568
[ 提交][ ][ ]

题目描述

营业额统计 Tiger最近被公司升任为营业部经理，他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。 Tiger拿出了公司的账本，账本上记录了公司成立以来每天的营业额。分析营业情况是一项相当复杂的工作。由于节假日，大减价或者是其他情况的时候，营业额会出现一定的波动，当然一定的波动是能够接受的，但是在某些时候营业额突变得很高或是很低，这就证明公司此时的经营状况出现了问题。经济管理学上定义了一种最小波动值来衡量这种情况： 该天的最小波动值 当最小波动值越大时，就说明营业情况越不稳定。 而分析整个公司的从成立到现在营业情况是否稳定，只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。 第一天的最小波动值为第一天的营业额。  输入输出要求

输入

第一行为正整数 ，表示该公司从成立一直到现在的天数，接下来的n行每行有一个正整数 ，表示第i天公司的营业额。

输出

输出文件仅有一个正整数，即Sigma(每天最小的波动值) 。结果小于2^31 。

样例输入

6
5
1
2
5
4
6

样例输出

12

提示

结果说明：5+|1-5|+|2-1|+|5-5|+|4-5|+|6-5|=5+4+1+0+1+1=12

题目：http://mail.bashu.cn:8080/bs_oj/showproblem?problem_id=2056

分析：这题算是一道水题了，不过数据有点错误，就是可能出现数据不到n个数的情况，这题很明显可以用线段树来做，不过由于开始学SBT，所以用SBT来搞了，模板题不解释~~~

代码：

#include<cstdio>
#include<iostream>
using namespace std;
const int mm=1111111;
const int mn=33333;
int L[mm],R[mm],S[mm],V[mm];
bool p[mm]={0};
int i,x,y,n,a,tt,root,ans;
void Right_Rotate(int &t)
{
    int k=L[t];
    L[t]=R[k];
    R[k]=t;
    S[k]=S[t];
    S[t]=S[L[t]]+S[R[t]]+1;
    t=k;
}
void Left_Rotate(int &t)
{
    int k=R[t];
    R[t]=L[k];
    L[k]=t;
    S[k]=S[t];
    S[t]=S[L[t]]+S[R[t]]+1;
    t=k;
}
void maintain(int &t,bool flag)
{
    if(flag)
        if(S[R[R[t]]]>S[L[t]])
            Left_Rotate(t);
        else if(S[L[R[t]]]>S[L[t]])
            Right_Rotate(R[t]),Left_Rotate(t);
        else return;
    else
        if(S[L[L[t]]]>S[R[t]])
            Right_Rotate(t);
        else if(S[R[L[t]]]>S[R[t]])
            Left_Rotate(L[t]),Right_Rotate(t);
        else return;
    maintain(L[t],0);
    maintain(R[t],1);
    maintain(t,0);
    maintain(t,1);
}
void Insert(int &t,int v)
{
    if(t)
    {
        ++S[t];
        if(v<V[t])Insert(L[t],v);
        else Insert(R[t],v);
        maintain(t,v>=V[t]);
    }
    else
    {
        V[t=++tt]=v;
        S[t]=1;
        L[t]=R[t]=0;
    }
}
int Pred(int t,int v)
{
    if(!t)return v;
    if(v<=V[t])return Pred(L[t],v);
    else
    {
        int tmp=Pred(R[t],v);
        return tmp==v?V[t]:tmp;
    }
}
int Succ(int t,int v)
{
    if(!t)return v;
    if(v>=V[t])return Succ(R[t],v);
    else
    {
        int tmp=Succ(L[t],v);
        return tmp==v?V[t]:tmp;
    }
}
int main()
{
    ans=tt=root=S[0]=0;
    scanf("%d%d",&n,&a);
    p[ans=a]=1;
    Insert(root,a);
    for(i=1;i<n;++i)
    {
        if(scanf("%d",&a)==EOF)a=0;
        if(!p[a])
        {
            if(!(x=a-Pred(root,a)))x=mm;
            if(!(y=Succ(root,a)-a))y=mm;
            ans+=min(x,y);
            Insert(root,a);
            p[a]=1;
        }
    }
    printf("%d\n",ans);
    return 0;
}

用splay解还是比较简单的，每次插入就把该节点旋转到根以保证树的平衡性。。。

附上splay代码：

#include<cstdio>
#include<iostream>
using namespace std;
const int mm=111111;
const int oo=1e9;
struct SplayTree
{
    int son[mm][2],far[mm],val[mm];
    int rt,size;
    void Link(int x,int y,int c)
    {//将x作为y的c儿子连接上
        far[x]=y,son[y][c]=x;
    }
    void Rotate(int x,int c)
    {//c=0则x是y的左儿子
        int y=far[x];
        Link(x,far[y],son[far[y]][1]==y);
        Link(son[x][!c],y,c);
        Link(y,x,!c);
    }
    void Splay(int x,int g)
    {
        for(;far[x]!=g;)
        {
            int y=far[x],cy=(son[far[y]][1]==y),cx=(son[y][1]==x);
            if(far[y]==g)Rotate(x,cx);
            else
            {
                if(cx==cy)Rotate(y,cy);
                else Rotate(x,cx);
                Rotate(x,cy);
            }
        }
        if(!g)rt=x;
    }
    void NewNode(int y,int &x,int a)
    {
        x=++size;
        far[x]=y,val[x]=a;
        son[x][0]=son[x][1]=0;
    }
    void Prepare()
    {
        NewNode(size=0,rt,-oo);
        NewNode(rt,son[rt][1],oo);
    }
    void Insert(int a)
    {
        int x=rt;
        while(son[x][val[x]<a])x=son[x][val[x]<a];
        NewNode(x,son[x][val[x]<a],a);
        Splay(size,0);
    }
    int Pre(int a)
    {
        int x=rt,ret=-oo;
        while(x)
        {
            if(val[x]==a)return a;
            if(val[x]<a)ret=max(ret,val[x]);
            x=son[x][val[x]<a];
        }
        return ret;
    }
    int Suc(int a)
    {
        int x=rt,ret=oo;
        while(x)
        {
            if(val[x]==a)return a;
            if(val[x]>a)ret=min(ret,val[x]);
            x=son[x][val[x]<a];
        }
        return ret;
    }
}spt;
int i,j,a,n,ans;
int main()
{
    scanf("%d%d",&n,&a);
    ans=a;
    spt.Prepare();
    spt.Insert(a);
    while(--n>0)
    {
        if(scanf("%d",&a)==-1)a=0;
        ans+=min(a-spt.Pre(a),spt.Suc(a)-a);
        spt.Insert(a);
    }
    printf("%d\n",ans);
    return 0;
}