题目:洛谷P2596、BZOJ1861、codevs1514。
题目大意:
给你一个数列(元素各不相同),有5个操作:
1. 把一个数提到最前面。
2. 把一个数提到最后面。
3. 把一个数和它前面/后面/自己(雾)交换。
4. 询问一个元素排第几。
5. 询问第X个元素的值。
解题思路:
维护一棵非旋Treap。
对于第4个操作,直接查询即可。
对于第5个操作,把那个元素分裂出来输出即可。
对于其他操作,每次把要改变的分裂出来,然后合并到目标位置即可。
C++ Code:
#include<bits/stdc++.h>
using namespace std;
int n,m,p,rt=0,s[800005],pos[800005];
char opt[10];
struct node{
int ls,rs,r,s,sz,fa;
bool rot;
}a[800005];
#define pd(x) if(a[x].rot)a[x].rot=0,a[a[x].ls].rot^=1,a[a[x].rs].rot^=1,a[x].ls^=a[x].rs^=a[x].ls^=a[x].rs
inline void update(int x){
a[x].sz=a[a[x].ls].sz+a[a[x].rs].sz+1;
a[a[x].ls].fa=x;
a[a[x].rs].fa=x;
}
int merge(int x,int y){
if(!x||!y)return x+y;
pd(x);pd(y);
if(a[x].r<a[y].r){
a[x].rs=merge(a[x].rs,y);
update(x);
return x;
}
a[y].ls=merge(x,a[y].ls);
update(y);
return y;
}
void split(int u,int k,int& x,int& y){
pd(u);
if(!k){
x=0,y=u;
return;
}
if(k==a[u].sz){
x=u,y=0;
return;
}
if(a[a[u].ls].sz>=k)split(a[u].ls,k,x,a[u].ls),y=u;else
split(a[u].rs,k-a[a[u].ls].sz-1,a[u].rs,y),x=u;
update(u);
}
int find(int u){
int siz=a[a[u].ls].sz+1;
for(;a[u].fa;u=a[u].fa)
if(u==a[a[u].fa].rs)siz+=a[a[a[u].fa].ls].sz+1;
return siz;
}
int main(){
srand(20170607);
int u,x,y,z,t,f;
ios::sync_with_stdio(0);
cin>>n>>m;
memset(a,0,sizeof a);
for(int i=1;i<=n;++i){
cin>>s[i];
pos[s[i]]=i;
a[i]=(node){0,0,rand(),s[i],1,0,0};
rt=merge(rt,i);
}
while(m--){
cin>>opt;
switch(opt[0]){
case'T':
cin>>p;
u=find(pos[p]);
split(rt,u-1,x,y);
split(y,1,y,z);
rt=merge(merge(y,x),z);
break;
case'B':
cin>>p;
u=find(pos[p]);
split(rt,u-1,x,y);
split(y,1,y,z);
rt=merge(merge(x,z),y);
break;
case'I':
cin>>p>>t;
if(t){
int u=find(pos[p]);
if(t==-1){
split(rt,u-2,x,y);
split(y,1,y,z);
split(z,1,z,f);
rt=merge(merge(merge(x,z),y),f);
}else{
split(rt,u-1,x,y);
split(y,1,y,z);
split(z,1,z,f);
rt=merge(merge(merge(x,z),y),f);
}
}
break;
case'A':
cin>>p;
cout<<find(pos[p])-1<<endl;
break;
case'Q':
cin>>p;
split(rt,p-1,x,y);
split(y,1,y,z);
cout<<a[y].s<<endl;
rt=merge(merge(x,y),z);
}
}
return 0;
}