【Tyvj1185】【codevs1296】【BZOJ1588】营业额统计，Splay练习

#include<bits/stdc++.h>
using namespace std;
int n,x,root,tot,p,q,ans,fuck;
struct os
{
int fa,occ,sz,ch[2],data;
}a[100000];
void ct(int now)
{
a[now].sz=a[now].occ+a[a[now].ch[0]].sz+a[a[now].ch[1]].sz;
}
void made(int x)
{
a[++tot].data=x;
a[tot].occ=a[tot].sz=1;
}
void rorate(int now,bool flag)
{
int pa=a[now].fa;
a[pa].ch[!flag]=a[now].ch[flag];
a[a[now].ch[flag]].fa=pa;
if (a[a[pa].fa].ch[0]==pa) a[a[pa].fa].ch[0]=now;
else a[a[pa].fa].ch[1]=now;
a[now].fa=a[pa].fa;
a[pa].fa=now;
a[now].ch[flag]=pa;
ct(pa);
ct(now);
}
void splay(int now,int goal)
{
int pa;
while (a[now].fa!=goal)
{
pa=a[now].fa;
if (a[pa].fa==goal)
{
if (a[pa].ch[0]==now) rorate(now,1);
else rorate(now,0);
}
else if (a[a[pa].fa].ch[0]==pa)
{
if (a[pa].ch[0]==now) rorate(pa,1);
else rorate(now,0);
rorate(now,1);
}
else
{
if (a[pa].ch[1]==now) rorate(pa,0);
else rorate(now,1);
rorate(now,0);
}
}
if (!goal) root=now;
}
void insert(int x)
{
if (!root) {made(x);root=tot;return;}
int now=root;
while (now)
{
a[now].sz++;
if (x==a[now].data){a[now].occ++;splay(now,0);return;}
else if (x<a[now].data)
{
if (!a[now].ch[0])
{
made(x);
a[now].ch[0]=tot;
a[tot].fa=now;
splay(tot,0);
return;
}
else now=a[now].ch[0];
}
else
{
if (!a[now].ch[1])
{
made(x);
a[now].ch[1]=tot;
a[tot].fa=now;
splay(tot,0);
return;
}
else now=a[now].ch[1];
}
}
}
int find(int x)
{
int now=root;
while (now)
{
if (a[now].data==x) return now;
if (a[now].data>x) now=a[now].ch[0];
else now=a[now].ch[1];
}
return 0;
}
int find_next_max(int x)
{
int now=root,ans=0;
while (now)
{
if (a[now].data>x)
ans=(!ans||a[ans].data>a[now].data)?now:ans,
now=a[now].ch[0];
else now=a[now].ch[1];
}
if (ans) splay(ans,0);
return ans;
}
int find_next_min(int x)
{
int now=root,ans=0;
while (now)
{
if (a[now].data<x)
ans=(!ans||a[ans].data<a[now].data)?now:ans,
now=a[now].ch[1];
else now=a[now].ch[0];
}
if (ans) splay(ans,0);
return ans;
}
main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
if (scanf("%d",&x)==EOF) x=0;
if (find(x)) {insert(x);continue;}
p=find_next_max(x);
q=find_next_min(x);
if (!p&&!q) ans+=x;
else if (!p) ans+=x-a[q].data;
else if (!q) ans+=a[p].data-x;
else ans+=(a[p].data-x>x-a[q].data)?x-a[q].data:a[p].data-x;
insert(x);
}
printf("%d",ans);
}