[ZJOI2006]书架
题目:洛谷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; }