bzoj1588营业额统计

Description

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

Input

第一行为正整数 ，表示该公司从成立一直到现在的天数，接下来的n行每行有一个整数(有可能有负数) ，表示第i
天公司的营业额。
天数n<=32767,每天的营业额ai <= 1,000,000。
最后结果T<=2^31

Output

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

Sample Input
6
5
1
2
5
4
6

Sample Output
12

分析：
以权值为下标建一颗线段树.
每次插一个新值, 并查询小于它的最大值和大于它的最小值.

这里写代码片
/*
权值线段树 
数组下标是权值，按顺序添加
*/

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long

using namespace std;

const int N=40000;
const int INF=0x33333333;
struct node{
    int x,y,mx,mn,v;
};
node tree[40000<<2];
int n,num[N];
ll ans=0;
struct node1{
    int bh,v;
};
node1 a[N];

int cmp(const node1 &a,const node1 &b)
{
    return a.v<b.v;
}

void update(int bh) //维护最大最小值
{
    int lc=bh<<1;
    int rc=bh<<1|1;
    if (tree[bh].x!=tree[bh].y)
    {
        tree[bh].mx=max(tree[lc].mx,tree[rc].mx);
        tree[bh].mn=min(tree[lc].mn,tree[rc].mn);
    }
    return;
}

void build(int bh,int l,int r)
{
    tree[bh].x=l;
    tree[bh].y=r;
    if (l==r) 
    {
        tree[bh].mx=-INF;
        tree[bh].mn=INF;
        return;
    }
    int mid=(l+r)>>1;
    build(bh<<1,l,mid);
    build(bh<<1|1,mid+1,r);
    update(bh);
}

void add(int bh,int mb,int v)
{
    if (tree[bh].x==mb&&tree[bh].y==mb) 
    {
        tree[bh].v=v;  //记录实际值
        tree[bh].mx=v;
        tree[bh].mn=v;
        return;
    }
    int mid=(tree[bh].x+tree[bh].y)>>1;  //这种优美的写法可以减少常数
    if (mb<=mid) add(bh<<1,mb,v);
    if (mb>mid) add(bh<<1|1,mb,v);
    update(bh);
}

int ask(int bh,int l,int r,int lx)
{  //把两种操作合并到一起，我可能是闲的
    if (r<l) 
    {
        if (lx==1) return -INF;
        else return INF;
    }
    if (tree[bh].x>=l&&tree[bh].y<=r)
    {
        if (lx==1)
        return tree[bh].mx;
        else return tree[bh].mn;
    }
    int mid=(tree[bh].x+tree[bh].y)>>1;
    int ans;
    if (lx==1) ans=-INF;
    else ans=INF;
    if (l<=mid)
    {
        if (lx==1)
        ans=max(ans,ask(bh<<1,l,r,lx));
        else ans=min(ans,ask(bh<<1,l,r,lx));
    } 
    if (r>mid) 
    {
        if (lx==1)
        ans=max(ans,ask(bh<<1|1,l,r,lx));
        else ans=min(ans,ask(bh<<1|1,l,r,lx));
    }
    return ans;
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i].v),a[i].bh=i;  //离散，真烦 
    sort(a+1,a+1+n,cmp);
    build(1,1,n);
    for (int i=1;i<=n;i++) num[a[i].bh]=i;  
    //离散化，位置是a[i].bh的数排序后排第i，离散后的值也是i
    add(1,num[1],a[num[1]].v);  //第一天的最小波动就是本身
    ans+=(ll)a[num[1]].v;
    for (int i=2;i<=n;i++)
    {
        add(1,num[i],a[num[i]].v);  
        int maxx=ask(1,1,num[i]-1,1); //小于ta的最大值
        int minn=ask(1,num[i]+1,n,2); //大于ta的最小值
        int t=min(abs(a[num[i]].v-minn),abs(maxx-a[num[i]].v));
        ans+=(ll)t; //开ll
    } 
    printf("%lld",ans);
    return 0;
}