您的位置:首页 > 其它

bzoj3223区间翻转

2016-06-23 20:24 393 查看
本着碰到平衡树都用可持久化treap的原则,找到了一个写的比较短的而且不用深坑的指针的写法,感觉写得还不错。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#define N 200005
#define mp make_pair
#define pa pair<int,int>
using namespace std;
struct treap{int val,ls,rs,sz,key,flip;}t
;
int n,m,root=0,num=0,cnt=0;
int v
;
void pushdown(int x)
{if(t[x].flip==0) return;
t[t[x].ls].flip=!t[t[x].ls].flip;
t[t[x].rs].flip=!t[t[x].rs].flip;
t[x].flip=0;
swap(t[x].ls,t[x].rs);
}
void pushup(int x) {t[x].sz=t[t[x].ls].sz+t[t[x].rs].sz+1;}
int merge(int a,int b)
{ if(a==0||b==0) return a+b;
if(t[a].key<t[b].key) {pushdown(a); t[a].rs=merge(t[a].rs,b);pushup(a);return a;}//要合并之前要先把标记下传
else{pushdown(b); t[b].ls=merge(a,t[b].ls);pushup(b); return b;}
}
pa split(int a,int k){
pa tmp;
if(k==0) return mp(0,a);
pushdown(a);int ls=t[a].ls,rs=t[a].rs;
if(k==t[ls].sz) {t[a].ls=0;pushup(a);return mp(ls,a);}//删除父子关系。。可以这么理解
if(k==t[ls].sz+1) {t[a].rs=0;pushup(a);return mp(a,rs);}
if(k<t[ls].sz) {tmp=split(ls,k);t[a].ls=tmp.second;pushup(a);return mp(tmp.first,a);}//这里是可持久化treap的核心。。。
if(k>t[ls].sz+1) {tmp=split(rs,k-t[ls].sz-1);t[a].rs=tmp.first;pushup(a);return mp(a,tmp.second);}
}
void dfs(int x){
if(x==0) return;
if(t[x].flip) pushdown(x);
dfs(t[x].ls);
v[++cnt]=x;
dfs(t[x].rs);
}
void ins(int &now,int x)
{ t[++num].val=x;//丝帛初始化
t[num].key=rand();
t[num].sz=1;
now=merge(now,num);

}
int main()
{scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) ins(root,i);//虽然插入这么写很慢
for(int i=1;i<=m;i++)
{int l,r;
scanf("%d%d",&l,&r);
pa tmp1=split(root,l-1);//这里就是提取出想要的根
pa tmp2=split(tmp1.second,r-l+1);
t[tmp2.first].flip^=1;
root=merge(tmp1.first,tmp2.first);
root=merge(root,tmp2.second);
}

dfs(root);
for(int i=1;i<=cnt;i++) printf("%d ",v[i]);puts("");//煞笔输出
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: