【BZOJ3678】wangxz与OJ
Description
某天,wangxz神犇来到了一个信息学在线评测系统(Online Judge)。由于他是一位哲♂学的神犇,所以他不打算做题。他发现这些题
目呈线性排列,被标记为1~n号,每道题都有一个难度值(可以<=0)。他决定与这些题目玩♂耍。
1、他可以在某个位置插♂入一些难度值特定的题目。
2、他可以吃♂掉(删除)一段题目。
3、他可以查询某个位置的题目的难度值。
维护一个初始有n个元素的序列(标记为1~n号元素),支持以下操作:
0 p a b (0<=p<=当前序列元素个数) (a<=b) 在p位置和p+1位置之间插入整数:a,a+1,a+2,...,b-1,b。若p为0,插在序列最前面;
1 a b (1<=a<=b<=当前序列元素个数) 删除a,a+1,a+2,...,b-1,b位置的元素;
2 p (1<=p<=当前序列元素个数) 查询p位置的元素。
Input
输入第一行包括两个正整数n(1<=n<=20000),m(1<=m<=20000),代表初始序列元素个数和操作个数。
接下来n个整数,为初始序列元素。
接下来m行,每行第一个为整数sym,
若sym=0,接下来有一个非负整数p,两个整数a,b;
若sym=1,接下来有两个正整数a,b;
若sym=2,接下来有一个正整数p;
p、x、y的含义及范围见题目描述。
在任何情况下,保证序列中的元素总数不超过100000。
保证题目涉及的所有数在int内。
Output
对每个sym=2,输出一行,包括一个整数,代表询问位置的元素。
Sample Input
5 3
1 2 3 4 5
0 2 1 4
1 3 8
2 2
Sample Output
2
题解:容易想到用Splay,我们的Splay的每个节点处维护的不是一个点,而是连续的一段数[a,b]。我们需要实现一个split操作:如果我们想将c和c+1分开,那么我们先找到c所在的节点,将它的后继提上来,并在二者之间新建一个点,将当前节点变成[a,c],并将新点变成[c+1,b]即可。特别地,如果c==b,则不用操作。
其余的操作就很容易了。
#include <cstdio> #include <iostream> #include <cstring> using namespace std; struct node { int ch[2],a,b,fa,siz; }s[200010]; int n,m,tot,rt; inline void pushup(int x) { s[x].siz=s[s[x].ch[0]].siz+s[s[x].ch[1]].siz+s[x].b-s[x].a+1; } int build(int l,int r) { if(l>r) return 0; int x=(l+r)>>1; s[x].ch[0]=build(l,x-1),s[x].ch[1]=build(x+1,r); if(s[x].ch[0]) s[s[x].ch[0]].fa=x; if(s[x].ch[1]) s[s[x].ch[1]].fa=x; pushup(x); return x; } inline void rotate(int x,int &k) { int y=s[x].fa,z=s[y].fa,d=(x==s[y].ch[1]); if(z) s[z].ch[y==s[z].ch[1]]=x; else if(y==k) k=x; s[x].fa=z,s[y].fa=x,s[y].ch[d]=s[x].ch[d^1]; if(s[x].ch[d^1]) s[s[x].ch[d^1]].fa=y; s[x].ch[d^1]=y; pushup(y),pushup(x); } inline void splay(int x,int &k) { while(x!=k) { int y=s[x].fa,z=s[y].fa; if(y!=k) { if((x==s[y].ch[0])^(y==s[z].ch[0])) rotate(x,k); else rotate(y,k); } rotate(x,k); } } int find(int x,int y) { if(!x) return 0; if(y>s[s[x].ch[0]].siz&&y<=s[s[x].ch[0]].siz+s[x].b-s[x].a+1) return x; if(s[s[x].ch[0]].siz>=y) return find(s[x].ch[0],y); return find(s[x].ch[1],y-s[s[x].ch[0]].siz-s[x].b+s[x].a-1); } inline int rd() { int ret=0,f=1; char gc=getchar(); while(gc<‘0‘||gc>‘9‘) {if(gc==‘-‘) f=-f; gc=getchar();} while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+(gc^‘0‘),gc=getchar(); return ret*f; } inline void split(int y) { int x=find(rt,y+1); splay(x,rt),splay(find(rt,s[s[x].ch[0]].siz+s[x].b-s[x].a+2),s[x].ch[1]); if(s[s[x].ch[0]].siz+s[x].b-s[x].a==y) return ; int z=++tot; s[z].b=s[x].b,s[x].b=s[x].a+y-s[s[x].ch[0]].siz,s[z].a=s[x].b+1; s[z].fa=s[x].ch[1],s[s[x].ch[1]].ch[0]=z; pushup(z),pushup(s[x].ch[1]),pushup(x); rotate(z,s[x].ch[1]); } int main() { n=rd(),m=rd(); int i,x,y,a,b,op; for(i=1;i<=n;i++) s[i+1].a=s[i+1].b=rd(); rt=build(1,n+2),tot=n+2; for(i=1;i<=m;i++) { op=rd(); if(op==0) { a=rd(),split(a); x=++tot,s[x].a=rd(),s[x].b=rd(),s[x].fa=s[rt].ch[1],s[s[rt].ch[1]].ch[0]=x; pushup(x),pushup(s[rt].ch[1]),pushup(rt); } if(op==1) { a=rd(),b=rd(),split(a-1),split(b); x=find(rt,a),splay(x,rt),y=find(rt,b+2),splay(y,s[x].ch[1]); s[y].ch[0]=0; pushup(y),pushup(x); } if(op==2) { a=rd(),x=find(rt,a+1),splay(x,rt); printf("%d\n",a-s[s[x].ch[0]].siz+s[x].a); } } return 0; }//5 3 1 2 3 4 5 0 2 1 4 1 3 8 2 2