TYVJ1185 营业额统计

也是旧题重做,本次是splay,原来写的treap版本请移步:http://www.cnblogs.com/evan-oi/archive/2012/02/03/2337324.html

 

也是splay水题不多说了。

 

View Code
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
#define MaxN 10010
#define INF 2000000
int n,tot=0,root=0,ans=0;
struct atp
{
int data,f,l,r,cnt;
}a[MaxN*8];

void Right(int x)
{
int y=a[x].f;
int z=a[y].f,c=a[x].r;
if (a[z].l==y) a[z].l=x; else a[z].r=x; a[x].f=z;
a[x].r=y;a[y].f=x;
a[y].l=c;a[c].f=y;
}
void Left(int x)
{
int y=a[x].f;
int z=a[y].f,c=a[x].l;
if (a[z].l==y) a[z].l=x; else a[z].r=x; a[x].f=z;
a[x].l=y;a[y].f=x;
a[y].r=c;a[c].f=y;
}
void splay(int &root,int x)
{
int y,z;
while (x!=root)
{
y=a[x].f;z=a[y].f;
if (y==root)
{
if (a[y].l==x) Right(x); else Left(x);
root=x;
} else
if (a[z].l==y)
if (a[y].l==x) Right(y),Right(x); else
Left(x),Right(x); else
if (a[y].r==x) Left(y),Left(x); else
Right(x),Left(x);
if (z==root) root=x;
}
}
void insert(int &t,int x,int f)
{
if (t==0)
{
tot++;
t=tot;
a[t].l=a[t].r=0;
a[t].f=f;
a[t].data=x;
a[t].cnt=1;
splay(root,t);
} else
if (x<a[t].data) insert(a[t].l,x,t); else
if(x>a[t].data) insert(a[t].r,x,t); else
if (x==a[t].data) a[t].cnt++,splay(root,t);
}
int Max(int t)
{
if (t==0) return INF;
if (a[t].r!=0) return Max(a[t].r); else
return a[t].data;
}
int Min(int t)
{
if (t==0) return INF;
if (a[t].l!=0) return Min(a[t].l); else
return a[t].data;
}
int main()
{
int x,t1,t2;
freopen("in","r",stdin);
//freopen("1.out","w",stdout);
scanf("%d",&n);
scanf("%d",&x);
ans=x;insert(root,x,0);
for (int i=2;i<=n;i++)
{
scanf("%d",&x);
insert(root,x,0);
if (a[root].cnt==1)
{
t1=Max(a[root].l);
t2=Min(a[root].r);
ans+=min(abs(x-t1),abs(x-t2));
}
}
printf("%d",ans);
return 0;
}



posted @ 2012-03-31 16:45  Evan1004  阅读(219)  评论(0编辑  收藏  举报