营业额统计

营业额统计

题目描述

---

核心思路

根据题意，最小波动值$f_i$其实就是要在$[1,i]$中找到一个$a_j$，使得$|a_i-a_j|$最小，那么$a_j$有可能比$a_i$大，也有可能比$a_j$小。意思就是说，我们要找到$\ge a_i$的最小数或者是$\le a_i$的最大数，这样才能保证$|a_i-a_j|$取到最小。因此，我们可以用平衡树的get_pre寻找前驱函数和get_next寻找后继函数。

问题：这里为什么是min(x-get_pre(root,x),get_next(root,x)-x)，不加绝对值呢？

首先get_pre(root,x)是求出小于等于x的最大数，设这个数为A，由于x大于等于A，因此x-A大于等于0，所以不用加绝对值；

其次get_next(root,x)-x是求出大于等于x的最小数，设这个数为B，由于x小于等于B，因此B-x大于等于0，所以不用加绝对值。

---

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=33010,INF=1e7;
int n;
struct Node{
    int l,r;
    int key,val;
}tr[N];
int root,idx;
void zig(int &p)
{
    int q=tr[p].l;
    tr[p].l=tr[q].r;
    tr[q].r=p;
    p=q;
}
void zag(int &p)
{
    int q=tr[p].r;
    tr[p].r=tr[q].l;
    tr[q].l=p;
    p=q;
}
int get_node(int key)
{
    tr[++idx].key=key;
    tr[idx].val=rand();
    return idx;
}
void build()
{
    get_node(-INF);
    get_node(INF);
    root=1;
    tr[1].r=2;
    if(tr[1].val<tr[2].val)
    zag(root);
}
void insert(int &p,int key)
{
    if(p==0)
    p=get_node(key);
    else if(tr[p].key==key)
    return;
    else if(tr[p].key>key)
    {
        insert(tr[p].l,key);
        if(tr[tr[p].l].val>tr[p].val)
        zig(p);
    }
    else
    {
        insert(tr[p].r,key);
        if(tr[tr[p].r].val>tr[p].val)
        zag(p);
    }
}
int get_pre(int p,int key)  //寻找小于等于key的最大数
{
    if(!p)
    return -INF;
    else if(tr[p].key>key)
    return get_pre(tr[p].l,key);
    else
    return max(tr[p].key,get_pre(tr[p].r,key));
}
int get_next(int p,int key) //寻找大于等于key的最小数
{
    if(!p)
    return INF;
    else if(tr[p].key<key)
    return get_next(tr[p].r,key);
    else
    return min(tr[p].key,get_next(tr[p].l,key));
}
int main()
{
    build();
    scanf("%d",&n);
    LL res=0;
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        if(i==1)
        res+=x;
        else
        res+=min(x-get_pre(root,x),get_next(root,x)-x);
        insert(root,x);
    }
    printf("%lld\n",res);
    return 0;
}

---