洛谷 P2234 [HNOI2002]营业额统计
Description
Tiger最近被公司升任为营业部经理,他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。
Tiger拿出了公司的账本,账本上记录了公司成立以来每天的营业额。分析营业情况是一项相当复杂的工作。由于节假日,大减价或者是其他情况的时候,营业额会出现一定的波动,当然一定的波动是能够接受的,但是在某些时候营业额突变得很高或是很低,这就证明公司此时的经营状况出现了问题。经济管理学上定义了一种最小波动值来衡量这种情况:
当最小波动值越大时,就说明营业情况越不稳定。
而分析整个公司的从成立到现在营业情况是否稳定,只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。
第一天的最小波动值为第一天的营业额。
该天的最小波动值=min{|该天以前某一天的营业额-该天营业额|}。
Input
输入由文件’turnover.in’读入。
第一行为正整数n(n<=32767) ,表示该公司从成立一直到现在的天数,接下来的n行每行有一个整数ai(|ai|<=1000000) ,表示第i天公司的营业额,可能存在负数。
Output
- 一个数,最小波动值之和。
Sample Input
6 5 1 2 5 4 6
Sample Output
12
题解:
- 平衡树。
- 找个前驱后继就行了… …
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#define N 40005
#define inf 0x3f3f3f3f
using namespace std;
struct T {int l, r, val, dat;} t[N];
int n, ans, root, tot, x, y, z;
int read()
{
int x = 0, f = 1; char c = getchar();
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') {x = x * 10 + c - '0'; c = getchar();}
return x *= f;
}
int New(int val)
{
t[++tot].val = val;
t[tot].dat = rand();
return tot;
}
void split(int p, int val, int &x, int &y)
{
if(!p) {x = y = 0; return;}
if(t[p].val <= val)
{
x = p;
split(t[p].r, val, t[p].r, y);
}
else
{
y = p;
split(t[p].l, val, x, t[p].l);
}
}
int merge(int x, int y)
{
if(!x || !y) return x + y;
if(t[x].dat > t[y].dat)
{
t[x].r = merge(t[x].r, y);
return x;
}
else
{
t[y].l = merge(x, t[y].l);
return y;
}
}
int preNext(int val, int op)
{
int p, ans;
if(!op)
{
split(root, val, x, y);
p = x; while(t[p].r) p = t[p].r;
}
else
{
split(root, val - 1, x, y);
p = y; while(t[p].l) p = t[p].l;
}
ans = t[p].val;
root = merge(x, y);
return ans;
}
void insert(int val)
{
split(root, val, x, y);
root = merge(merge(x, New(val)), y);
}
int main()
{
New(-inf), New(inf), root = merge(1, 2);
cin >> n;
ans = read(), insert(ans);
for(int i = 2; i <= n; i++)
{
int val = read();
int v1 = preNext(val, 0);
int v2 = preNext(val, 1);
if(v1 == -inf) ans += abs(v2 - val);
else if(v2 == inf) ans += abs(v1 - val);
else ans += min(abs(v1 - val), abs(v2 - val));
insert(val);
}
cout << ans;
return 0;
}