BZOJ 1588: [HNOI2002]营业额统计

普通的，普通的平衡树，普通的小水题

用set水过（不过据说数据有问题。。。读入有一组少一行）

#include<cstdio>
#include<cstring>
#include<set>
#define inf 0x3f3f3f3f
using namespace std;

set<int> tree;
int read(){
	char ch=getchar();
	int x=0,f=1;
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
int x,n,m,ans;
int main(){
	n=read();
	tree.insert(-inf);
	tree.insert(inf);
	x=read();
	tree.insert(x);
	ans=x;
	for(int i=1;i<n-1;++i)
	{
		x=read();
		int l=*--tree.lower_bound(x),r=*tree.lower_bound(x);
		if(x-l>r-x)ans+=r-x;else ans+=x-l;
		tree.insert(x);
	}
	if(scanf("%d",&x)==1)
	{
		int l=*--tree.lower_bound(x),r=*tree.lower_bound(x);
		if(x-l>r-x)ans+=r-x;else ans+=x-l;
	}
	else
	{
		x=0;
		int l=*--tree.lower_bound(x),r=*tree.lower_bound(x);
		if(x-l>r-x)ans+=r-x;else ans+=x-l;
	}
	printf("%d",ans);
	return 0;
}

这是用set的

后来写了个treap

treap利用随机数来判定要不要旋转，对生成的随机数有堆的性质。

神犇叫它terap。。。同理还有被叫成spaly的splay。。

#include<iostream>
#include<cstdio>
#include<algorithm>
#define inf 0x3f3f3f3f
#define maxn 50100
using namespace std;
int s[maxn],l[maxn],r[maxn],v[maxn],rnd[maxn];
int n,sz,ans1,ans2,ans,root;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void update(int k){s[k]=s[l[k]+s[r[k]]]+1;}
void rt(int &k)      //右旋
{int t=l[k];l[k]=r[t];r[t]=k;s[t]=s[k];update(k);k=t;}
void lt(int &k)      //左旋
{int t=r[k];r[k]=l[t];l[t]=k;s[t]=s[k];update(k);k=t;}
void insert(int &k,int x){        //插入新的节点
    if(!k){k=++sz;v[k]=x;s[k]=1;rnd[k]=rand();return;}
    s[k]++;
    if(x<v[k]){insert(l[k],x);if(rnd[l[k]]<rnd[k])rt(k);}    //如果随机数不满足小根堆的性质就旋转
    else{insert(r[k],x);if(rnd[r[k]]<rnd[k])lt(k);}
}
void pred(int k,int x){           //求前驱
    if(!k)return;
    if(x>=v[k])ans1=v[k],pred(r[k],x);
    else pred(l[k],x);
}
void succ(int k,int x){           //求后继
    if(!k)return;
    if(x<=v[k])ans2=v[k],succ(l[k],x);
    else succ(r[k],x);
}
int main(){
    n=read();
    int x=read();
    insert(root,x);
    ans=x;
    for(int i=2;i<n;++i){
        int x=read();
        ans1=-inf,ans2=inf;
        pred(root,x);
        succ(root,x);
        ans+=min(x-ans1,ans2-x);
        insert(root,x);
    }
    if(scanf("%d",&x)==EOF)x=0;
    ans1=-inf,ans2=inf;
    pred(root,x);
    succ(root,x);
    ans+=min(x-ans1,ans2-x);
    cout<<ans<<endl;
}

刷了这么多遍居然比set还要慢。。。。。。看来是我写的不好。。。

2016.8.7

用替罪羊树又写了一遍。。

替罪羊树很快啊，而且好写。。具体说明看这篇 BZOJ 1588:[HNOI2002]营业额统计，这里就贴个代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define g getchar()
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
inline ll read(){
    ll x=0,f=1;char ch=g;
    for(;ch<'0'||ch>'9';ch=g)if(ch=='-')f=-1;
    for(;ch>='0'&&ch<='9';ch=g)x=x*10+ch-'0';
    return x*f;
}
inline void out(ll x){
    int a[25],t=0;
    if(x<0)putchar('-'),x=-x;
    for(;x;x/=10)a[++t]=x%10;
    for(int i=t;i;--i)putchar('0'+a[i]);
    if(t==0)putchar('0');
    putchar('\n');
}
int sz,size[50005],t[50005][2],fa[50005],v[50005],id,q[50005],top,rt,ans1,ans2,n,ans;
int ins(int &k,int x,int f){
    if(!k){k=++sz;size[k]=1;fa[k]=f;return 0;}
    if(x<v[k])ins(t[k][0],x,k);else ins(t[k][1],x,k);
    if(max(size[t[k][0]],size[t[k][1]])>size[k]*0.7)id=k;//如果不平衡，记录要重构的子树的根
    size[k]=size[t[k][0]]+size[t[k][1]]+1;//size记录节点的重量
}
int build(int &k,int l,int r){
    if(l>r)return 0;
    int mid=(l+r)>>1;
    k=q[mid];
    build(t[k][0],l,mid-1);
    build(t[k][1],mid+1,r);
    fa[t[k][0]]=fa[t[k][1]]=k;
    size[k]=size[t[k][0]]+size[t[k][1]]+1;
}//构建
void dfs(int x){if(!x)return;dfs(t[x][0]);q[++top]=x;dfs(t[x][1]);}//求dfs序
int re_build(int x){
    int pre=fa[x];
    dfs(x);
    for(int i=1;i<=top;++i)
    t[q[i]][0]=t[q[i]][1]=fa[q[i]]=size[q[i]]=0;
    if(!pre){build(rt,1,top);}//如果x是总树的根，重构总树
    else
    if(t[pre][0]==x)build(t[pre][0],1,top);
    else build(t[pre][1],1,top);
    fa[t[pre][0]]=fa[t[pre][1]]=pre;
    top=id=0;
}//重构以x为根的树
void lower(int k,int x){
    if(!k)return;if(v[k]<=x){ans1=v[k];lower(t[k][1],x);}else lower(t[k][0],x);
}
void higher(int k,int x){
    if(!k)return;if(v[k]>=x){ans2=v[k];higher(t[k][0],x);}else higher(t[k][1],x);
}

int main(){
//  freopen("turnover0.in","r",stdin);
//  freopen("","w",stdout);
    n=read();
    for(int i=1;i<=n;++i){
        int x;if(scanf("%d",&x)==EOF)x=0;
        ans1=-inf;ans2=inf;
        lower(rt,x);higher(rt,x);
        ans+=i==1?x:min(x-ans1,ans2-x);
        v[i]=x;ins(rt,x,0);
        if(id)re_build(id);
    }
    out(ans);
    return 0;
}