hdu 2475 BOX (splay)

版权声明：本文为博主原创文章，未经博主允许不得转载。

hdu 2475

Splay树是一种神奇的东西。。。

题意：

　　有一些箱子，要么放在地上，要么放在某个箱子里面 。

　　现在有两种操作：

　　　　（1） MOVE x y： 把 x 号箱子放到 y 号箱子里面，操作不合法就忽略这一次操作 。

　　　　（2） QUERY x :  查询 x 号箱子最外面的箱子是哪一个

解法：

　　首先对所有的树进行一遍DFS，得到一个DFS序，可以把它看成是一个括号序列，开始访问某个节点是左括号，结束访问时是右括号 。这样这题就转换成用Splay树来维护这个序列 。

　　对于MOVE操作：

　　　　将 x 对应那一段序列切下来，并将其放到 y 对应左括号的右边 。

　　对于QUERY操作：

　　　　直接查询以 x 为根的子树的最小值 。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <map>
#define ll long long

using namespace std;

const int N=50007;

//  前向星
struct edge{
    int to,nex;
}e[N];
int cnt;
int head[N];

inline void addedge(int u,int v){
    e[cnt].to=v; 
    e[cnt].nex=head[u];
    head[u]=cnt++;
}

int id[N*2];   // DFS序
int cou;

int tot;
struct tree{
    int key,fa,son[2];
}t[N*2];
int L[N],R[N];  // 第i个箱子所对应的左右括号在Splay树上的序号

inline void pushup(int x){}
inline void pushdown(int x){}

inline void rotate(int x,int p){
    int y=t[x].fa;
    pushdown(y);
    pushdown(x);

    t[y].son[!p]=t[x].son[p];
    t[t[x].son[p]].fa=y;
    t[x].fa=t[y].fa;
    if (t[x].fa)
        t[t[x].fa].son[t[t[x].fa].son[1]==y]=x;
    t[x].son[p]=y;
    t[y].fa=x;
    pushup(y);
    pushup(x);
}

inline void Splay(int x,int to){
    while (t[x].fa!=to){
        if (t[t[x].fa].fa==to)
            rotate(x,t[t[x].fa].son[0]==x);
        else {
            int y=t[x].fa, z=t[y].fa;
            int p=(t[z].son[0]==y);
            if (t[y].son[p]==x) 
                rotate(x,!p),rotate(x,p);
            else
                rotate(y,p),rotate(x,p);
        }
    }
}

inline int newnode(int key,int fa){
    int x=tot++;
    t[x].key=key;
    t[x].fa=fa;
    t[x].son[0]=t[x].son[1]=0;

    if (key>0) L[key]=x;
    else R[-key]=x;

    return x;
}

inline int get_min(int x){
    while (t[x].son[0]!=0)
        x=t[x].son[0];
    return x;
}

inline int get_max(int x){
    while (t[x].son[1]!=0) 
        x=t[x].son[1];
    return x;
}

inline int bulid(int l,int r,int fa){
    if (l>r) return 0;
    int x,mid=(l+r)>>1;
    x=newnode(id[mid],fa);
    t[x].son[0]=bulid(l,mid-1,x);
    t[x].son[1]=bulid(mid+1,r,x);
    pushup(x);
    return x;
}

inline void dfs(int u){    
    id[cou++]=u;
    for (int i=head[u];~i;i=e[i].nex)
        dfs(e[i].to);
    id[cou++]=-u;
}

inline void init(){
    memset(head,-1,sizeof(head));
    cnt=0;
    cou=0;
    tot=1;
}

inline int query(int a){
    int x=L[a];
    Splay(x,0);
    x=get_min(x);
    return t[x].key;
}

inline void move(int a,int b){
    if (a==b) return;

    int x=L[a],y=R[a];
    Splay(x,0);
    Splay(y,x);

    int xx=t[x].son[0],yy=t[y].son[1],z=0;
    z=get_max(xx);

    t[x].son[0]=0; t[y].son[1]=0;
    t[xx].fa=0; t[yy].fa=0;
    if (z!=0) t[z].son[1]=yy;
    t[yy].fa=z;

    if (b==0) return;

    if (query(b)==a){
        t[x].son[0]=xx; t[y].son[1]=yy;
        t[xx].fa=x; t[yy].fa=y;
        t[z].son[1]=0;
        return;
    }

    int l=L[b],r;
    Splay(l,0);
    r=get_min(t[l].son[1]);
    Splay(r,l);
    t[r].son[0]=x;
    t[x].fa=r;
}

int main(){
    int n,q;
    int x,y;
    char ch[10];
    bool f=false;
    while (scanf("%d",&n)!=EOF){
        if (f) puts("");
        else f=true;

        init();
        
        for (int i=1;i<=n;i++){
            scanf("%d",&x);
            addedge(x,i);
        }

        dfs(0);
        int k=0,st=1;
        for (int i=1;i<=2*n;i++){
            if (id[i]>0) k++;
            else k--;
            if (k==0) {
                bulid(st,i,0);
                st=i+1;
            }
        }

        scanf("%d",&q);
        while (q--){
            scanf("%s",ch);
            if (ch[0]=='Q') {
                scanf("%d",&x);
                printf("%d
",query(x));
            }
            else {
                scanf("%d %d",&x,&y);
                move(x,y);
            }
        }
    }

    return 0;
}