洛谷p2234/BZOJ1588 [HNOI2002]营业额统计
分析:###
好像没有什么好说的就是一个平衡树的板子……唯一要注意的就是这里要找的并不是严格的前驱和后继,因为如果找到之前某一天的营业额和它相等那么差就是0,所以我们仍然在结构体中开一个域cnt来存储同一个元素存储了多少次,如果a[p].cnt>1说明这个元素已经出现了不止一次了,那么直接跳出循环,返回a[p].val即可。
这一段代码贴在这里:
if(val==a[p].val){
if(a[p].cnt>1){
ans=p;
break;
}
...
}
然后说一下我的沙雕错误……建树的时候手一抽在左子树上压了个INF,在右子树上压了个-INF,然后敲敲打打找了两个小时的bug……
对于这件事我只想说:妈的智障!
全部代码如下:
#include<bits/stdc++.h>
#define maxn 40000
using namespace std;
struct treap{
int val;int l,r;int dat;int size;int cnt;
}a[maxn];
int tot,root,n,inf=0x7fffffff,ans=0;
inline int read(){
int cn=0,f=1;char c;
c=getchar();
while(!isdigit(c)){
if(c=='-')f=-1;
c=getchar();
}
while(isdigit(c)){
cn=cn*10+c-'0';
c=getchar();
}
return cn*f;
}
inline int New(int val){
a[++tot].val=val;
a[tot].dat=rand();
a[tot].cnt=a[tot].size=1;
return tot;
}
inline void update(int p){
a[p].size=a[a[p].l].size+a[a[p].r].size+a[p].cnt;
}
inline void build_tree(){
New(-inf),New(inf);
root=1;a[1].r=2;
update(root);
}
void zig(int &p){
int q=a[p].l;
a[p].l=a[q].r,a[q].r=p,p=q;
update(a[p].r),update(p);
}
void zag(int &p){
int q=a[p].r;
a[p].r=a[q].l,a[q].l=p,p=q;
update(a[p].l),update(p);
}
void insert(int &p,int val){
if(p==0){
p=New(val);
return;
}
if(val==a[p].val){
a[p].cnt++,update(p);
return;
}
if(val<a[p].val){
insert(a[p].l,val);
if(a[p].dat<a[a[p].l].dat)zig(p);
}
else{
insert(a[p].r,val);
if(a[p].dat<a[a[p].r].dat)zag(p);
}
update(p);
}
int get_pre(int val){
int ans=1;//a[1].val==-inf
int p=root;
while(p){
if(val==a[p].val){
if(a[p].cnt>1){
ans=p;
break;
}
if(a[p].l>0){
p=a[p].l;
while(a[p].r>0)p=a[p].r;
ans=p;
}
break;
}
if(a[p].val<val&&a[p].val>a[ans].val) ans=p;
p=val<a[p].val?a[p].l:a[p].r;
}
return a[ans].val;
}
int get_next(int val){
int ans=2;// a[2].val==inf
int p=root;
while(p){
if(val==a[p].val){
if(a[p].cnt>1){
ans=p;
break;
}
if(a[p].r>0){
p=a[p].r;
while(a[p].l>0)p=a[p].l;
ans=p;
}
break;
}
if(a[p].val>val&&a[p].val<a[ans].val) ans=p;
p=val<a[p].val?a[p].l:a[p].r;
}
return a[ans].val;
}
int main(){
// freopen("turnover.in","r",stdin);
// freopen("turnover.out","w",stdout);
n=read();
build_tree();
srand(19260817);
for(register int i=1;i<=n;i++){
int x;
x=read();
insert(root,x);
if(i==1)
ans+=x;
else{
if(get_pre(x)==-inf){
ans+=get_next(x)-x;
continue;
}
if(get_next(x)==inf){
ans+=x-get_pre(x);
continue;
}
ans+=min(x-get_pre(x),get_next(x)-x);
}
// cout<<ans<<endl;
}
printf("%d",ans);
return 0;
}