HNOI2002 营业额统计（Splay 找前驱 后继）

题目描述

Tiger最近被公司升任为营业部经理，他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。

Tiger拿出了公司的账本，账本上记录了公司成立以来每天的营业额。分析营业情况是一项相当复杂的工作。由于节假日，大减价或者是其他情况的时候，营业额会出现一定的波动，当然一定的波动是能够接受的，但是在某些时候营业额突变得很高或是很低，这就证明公司此时的经营状况出现了问题。经济管理学上定义了一种最小波动值来衡量这种情况：

当最小波动值越大时，就说明营业情况越不稳定。

而分析整个公司的从成立到现在营业情况是否稳定，只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。

第一天的最小波动值为第一天的营业额。

输入输出格式

输入格式：

第一行为正整数n（n<=32767） ，表示该公司从成立一直到现在的天数，接下来的n行每行有一个整数ai(|ai|<=1000000) ，表示第i天公司的营业额，可能存在负数。

输出格式：

输出文件仅有一个正整数，即每天最小波动值之和，小于2^31

输入输出样例

输入样例#1：

6
5
1
2
5
4
6

输出样例#1：

12

说明

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

思路 + 代码

这道题读懂了就算不知道怎么写也只知道要求什么了吧

就是要去找输入数的前驱（比它小的数的最大值）和后继（比它大的数的最小值）

当然找前驱和后继是有STL的，有兴趣可以自己搜一下相关的知识，我这里就放一下代码（拿的hh大佬的代码）

#include<set>
#include<algorithm>
#include<cmath>
using namespace std;
set <int> s;
int main()
{
    int n;
    scanf("%d",&n);
    int sum=0;
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        if(i==1)
        {
            s.insert(x);
            sum+=x;
        }
        else
        {
            set <int> ::iterator a=s.lower_bound(x);
            set <int> ::iterator b=a;
            if(a!=s.begin()) a--;
            if(b==s.end()) b--;
            sum+=min(abs(x-*a),abs(x-*b));
            s.insert(x);
        }
    }
    printf("%d",sum);
    return 0;
}

我自己是用Splay写的，为什么是这么旋转看看下面的样例再自己推一下就知道了

其他的函数注释写的很详细了就不多说了

#include<cstdio>
#include<algorithm>
using namespace std;
struct node{
    int fa,ld,rd,data; //父节点，左孩子，右孩子，值 
}tree[33000];
int n,ans,root,flag;
void rightx(int x) //右旋 
{
    int y=tree[x].fa;
    int z=tree[y].fa;
    tree[y].ld=tree[x].rd;
    if(tree[x].rd) tree[tree[x].rd].fa=y;
    tree[x].fa=z;
    if(z)
    {
        if(tree[z].ld==y)
        tree[z].ld=x;
        else tree[z].rd=x;
    } 
    tree[x].rd=y;
    tree[y].fa=x;
}
void leftx(int x) //左旋 
{
    int y=tree[x].fa;
    int z=tree[y].fa;
    tree[y].rd=tree[x].ld;
    if(tree[x].ld) tree[tree[x].ld].fa=y;
    tree[x].fa=z;
    if(z)
    {
        if(tree[z].ld==y)
        tree[z].ld=x;
        else tree[z].rd=x;
    }
    tree[x].ld=y;
    tree[y].fa=x;
}
int aabs(int x)
{
    return x>0?x:-x;
}//自己写的绝对值函数，不喜欢用cmath 
void splay(int x)
{
    while(tree[x].fa) //一直把x旋转到根节点 
    {
        int y=tree[x].fa; //y是x的父节点 
        int z=tree[y].fa; //z是y的父节点 
        if(!z)            //如果y是根的话 
        {
            if(tree[y].ld==x) rightx(x); //左孩子就右旋 
            else leftx(x);               //右孩子就左旋 
        }
        else
        {
            if(tree[y].ld==x&&tree[z].ld==y) //如果是 / 这种情况 
            rightx(y),rightx(x);

            if(tree[y].rd==x&&tree[z].rd==y) //如果是 \ 这种情况 
            leftx(y),leftx(x);

            //如果是祖孙关系是一条直线上先旋转父节点再旋转子节点 

            if(tree[y].rd==x&&tree[z].ld==y) //如果是 < 这种情况 
            leftx(x),rightx(x);

            if(tree[y].ld==x&&tree[z].rd==y) //如果是 > 这种情况 

            //如果是祖孙关系是一条折线上只需要旋转两次子节点
            //旋转的过程自己推一下就知道为什么这么旋转了
        }
    }
    root=x; //更新根节点 
}
void BST(int x,int rt) //建立二叉查找树 
{
    if(tree[x].data==tree[rt].data) 
    //相同的数只需要更新根节点没必要加入到树里去
    //因为相同的数有一个就够了 
    {
        splay(rt);
        flag=1;
        return;
    } //可以不要，只是为了简化时间复杂度 
    if(tree[x].data<tree[rt].data)
    {
        if(tree[rt].ld==0)
        {
            tree[rt].ld=x;
            tree[x].fa=rt;
        }
        else BST(x,tree[rt].ld);
    }
    else
    {
        if(tree[rt].rd==0)
        {
            tree[rt].rd=x;
            tree[x].fa=rt;
        }
        else BST(x,tree[rt].rd);
    }
}
int fro(int x) //找前驱节点
{
    int y=tree[x].ld;
    if(!y) return y;
    while(tree[y].rd)
    y=tree[y].rd;
    return y;
}
int beh(int x) //找后继节点
{
    int y=tree[x].rd;
    if(!y) return y;
    while(tree[y].ld)
    y=tree[y].ld;
    return y;
}
void insert(int x) //输入 
{
    flag=0;
    BST(x,root);  //建立二叉查找树 
    if(flag==1) return; //如果它有相同没必要计算 
    splay(x);

    int q=fro(x),h=beh(x),minn=0x7fffffff;
    if(q) minn=min(minn,aabs(tree[x].data-tree[q].data));
    if(h) minn=min(minn,aabs(tree[x].data-tree[h].data));

    ans+=minn;
}
int main()
{
    scanf("%d",&n);
    scanf("%d",&tree[1].data);
    root=1;ans+=tree[1].data;
    for(int i=2;i<=n;i++)
    {
        scanf("%d",&tree[i].data);
        insert(i);
    }
    printf("%d",ans);
    return 0;
}