营业额统计 Treap TYVJ1185
题目连接:http://tyvj.cpwz.cn/Problem_Show.asp?id=1185
相关知识:平衡树
题目不再赘述,比较简单的平衡树入门题,而且不用删除查询操作,只要直接要用数据结构模块就可以,在插入的时候更新答案,因为插入也是慢慢向着插入数值大小走的。
View Code
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<iostream>
using namespace std;
#define INF 20000000
#define MaxN 40000
struct treap
{
int x,p,l,r;
}a[MaxN];
int n,tot,ans,root,Max;
void turnright(int &t)
{
int t1=a[t].l;
a[t].l=a[t1].r;
a[t1].r=t;
t=t1;
}
void turnleft(int &t)
{
int t1=a[t].r;
a[t].r=a[t1].l;
a[t1].l=t;
t=t1;
}
void insert(int &t,int x)
{
if (t==0)
{
tot++;
t=tot;
a[t].l=0;
a[t].r=0;
a[t].x=x;
a[t].p=rand();
} else
{
if (abs(a[t].x-x)<Max) Max=abs(a[t].x-x);
if (x<a[t].x)
{
insert(a[t].l,x);
if (a[a[t].l].p < a[t].p) turnright(t);
}
else if (x>a[t].x)
{
insert(a[t].r,x);
if (a[a[t].r].p < a[t].p) turnleft(t);
}
}
}
int main()
{
int x;
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
Max=INF;
scanf("%d",&x);
insert(root,x);
if (i==1) ans+=x; else ans+=Max;
}
printf("%d\n",ans);
// system("pause");
return 0;
}