[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;
}

 

posted @ 2018-04-09 20:14  Mrsrz  阅读(189)  评论(0编辑  收藏  举报