Hdu 1890 Robotic Sort(Spaly 区间翻转)

传送门:Hdu 1890 Robotic Sort

---

题意:给你一个序列,每次输出第i小的数所在的位置,并将i-这个位置之间的数翻转

---

思路:Splay的区间翻转,利用一个lazy标记,按下标的中序遍历建树,然后刚开始的每个a[i]都有一个对应的下标,把这个下标存下来,
然后对a[i]排序,第i大的数刚开始对应的下标就是a[i].id,只要找出现在这个a[i].id在Splay中对应的位置就好了
每次操作都把最上面的根删除就行了

需要注意的一点是应该先pushdown在旋转,这样才能保证旋转的正确性

---

#include<bits/stdc++.h>
using namespace std;
const int N=100100;
struct node{
    int num,id;
}a[N];
int root,siz[N],ch[N][2],fa[N],rev[N];;//分别表示根,子树结点数量,左右孩子(0为左孩子,1为右孩子),翻转标记

bool cmp(node u,node v){
    return u.num==v.num ? u.id<v.id :u.num<v.num;
}

void Newnode(int &now,int father,int m){
    now=m;
    fa[now]=father;
    ch[now][0]=ch[now][1]=rev[now]=0;
    siz[now]=1;
}

void pushup(int x){
    siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
}

void build(int &x,int l,int r,int father){
    if(l>r)
        return ;
    int m=(l+r)>>1;
    Newnode(x,father,m);
    build(ch[m][0],l,m-1,m);
    build(ch[m][1],m+1,r,m);
    pushup(m);
}

//旋转，kind为1为右旋，kind为0为左旋
void Rotate(int x,int kind){
    int y=fa[x];
    ch[y][!kind]=ch[x][kind];
    fa[ch[x][kind]]=y;
    //如果父节点不是根结点，则要和父节点的父节点连接起来
    if(fa[y])
        ch[ fa[y] ][ch[fa[y]][1]==y]=x;
    fa[x]=fa[y];
    ch[x][kind]=y;
    fa[y]=x;
    pushup(y);
}

void pushdown(int now){
    if(rev[now]){
        rev[ch[now][0]]^=1;
        rev[ch[now][1]]^=1;
        swap(ch[now][0],ch[now][1]);
        rev[now]=0;
    }
}
//Splay调整，将根为now的子树调整为goal
void Splay(int now,int goal){
    pushdown(now);
    while(fa[now]!=goal){
        pushdown(fa[fa[now]]),pushdown(fa[now]),pushdown(now);
        if(fa[ fa[now] ]==goal)
            Rotate(now,ch[ fa[now] ][0]==now);
        else{
            int pre=fa[now],kind=ch[ fa[pre] ][0]==pre; //左儿子为1,右儿子为0
            if(ch[pre][kind]==now){ //两个方向不同
                Rotate(now,!kind);
                Rotate(now,kind);
            }
            else{   //两个方向相同
                Rotate(pre,kind);
                Rotate(now,kind);
            }
        }
    }
    if(goal==0) root=now;
    pushup(now);
}

int getmax(int x){
    pushdown(x);
    while(ch[x][1])
        x=ch[x][1],pushdown(x);
    return x;
}

void remove(){
    if(ch[root][0]==0){
        root=ch[root][1];
        fa[root]=0;
    }
    else{
        int m=getmax(ch[root][0]);
        Splay(m,root);
        ch[m][1]=ch[root][1];
        fa[ch[root][1]]=m;
        root=m;
        fa[root]=0;
        pushup(root);
    }
}

void init(int n){
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i].num),a[i].id=i;
    sort(a+1,a+n+1,cmp);
    root=0;
    ch[0][0]=ch[0][1]=siz[0]=0;
    build(root,1,n,0);
}

void Treaval(int x) {
    if(x) {
        Treaval(ch[x][0]);
        printf("结点%2d:左儿子 %2d 右儿子 %2d 父结点 %2d size = %2d\n",x,ch[x][0],ch[x][1],fa[x],siz[x]);
        Treaval(ch[x][1]);
    }
}
void debug() {printf("%d\n",root);Treaval(root);}

int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
        if(n==0)
            break;
        init(n);
        //debug();
        for(int i=1;i<n;i++){
            Splay(a[i].id,0);
            //debug();
            rev[ch[root][0]]^=1;
            printf("%d ",i+siz[ch[root][0]]);
            remove();//删掉根
        }
        printf("%d\n",n);
    }
    return 0;
}