HDU 1890 Robotic Sort [splay]

题意：给出一个序列，第 i 次操作把当前序列 第i位 至 第i小所在的位置 构成的区间进行翻转操作，这样i次后最后序列就变成有序的啦！问每次操作时第i小所在的位置在哪。

范围：N<=10W

解法：splay模板题，每次通过find-kth操作找到第i位，以及第i 小所在的位置，spaly至顶端获取这段区间，打上标记即可，注意splay之前必须pushdown

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<iostream>
#include<stdlib.h>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<bitset>
#pragma comment(linker, "/STACK:1024000000,1024000000")
template <class T>
bool scanff(T &ret){ //Faster Input
    char c; int sgn; T bit=0.1;
    if(c=getchar(),c==EOF) return 0;
    while(c!='-'&&c!='.'&&(c<'0'||c>'9')) c=getchar();
    sgn=(c=='-')?-1:1;
    ret=(c=='-')?0:(c-'0');
    while(c=getchar(),c>='0'&&c<='9') ret=ret*10+(c-'0');
    if(c==' '||c=='\n'){ ret*=sgn; return 1; }
    while(c=getchar(),c>='0'&&c<='9') ret+=(c-'0')*bit,bit/=10;
    ret*=sgn;
    return 1;
}
#define inf 1073741823
#define llinf 4611686018427387903LL
#define PI acos(-1.0)
#define lth (th<<1)
#define rth (th<<1|1)
#define rep(i,a,b) for(int i=int(a);i<=int(b);i++)
#define drep(i,a,b) for(int i=int(a);i>=int(b);i--)
#define gson(i,root) for(int i=ptx[root];~i;i=ed[i].next)
#define tdata int testnum;scanff(testnum);for(int cas=1;cas<=testnum;cas++)
#define mem(x,val) memset(x,val,sizeof(x))
#define mkp(a,b) make_pair(a,b)
#define findx(x) lower_bound(b+1,b+1+bn,x)-b
#define pb(x) push_back(x)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;

#define NN 200200
int c[NN][2],fa[NN],sz[NN],val[NN],a[NN];
bool rev[NN];
inline void update(int x){
    sz[x]=sz[c[x][0]]+sz[c[x][1]]+1;
}
inline void push(int x){
    if(rev[x]){
        rev[x]^=1;rev[c[x][0]]^=1;rev[c[x][1]]^=1;
        swap(c[x][0],c[x][1]);
    }
}
void rotate(int x,int &k){
    int y=fa[x],z=fa[y],l,r;
	l=(c[y][1]==x);r=l^1;
	if(y==k)k=x;
	else c[z][c[z][1]==y]=x;
	fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
	c[y][l]=c[x][r];c[x][r]=y;
	update(y);update(x);
}
void splay(int x,int &k){
    while(x!=k){
        int y=fa[x],z=fa[y];
        if(y!=k){
            if(c[y][0]==x^c[z][0]==y)rotate(x,k);
            else rotate(y,k);
        }
        rotate(x,k);
    }
}
void built(int l,int r,int f){
    if(l>r)return;
    int m=(l+r)>>1;
    c[f][m>=f]=m;
    fa[m]=f;
    sz[m]=1;
    val[m]=a[m];
    if(l==r)return;
    built(l,m-1,m);
    built(m+1,r,m);
    update(m);
}
int findk(int x,int k){
    push(x);
    int l=c[x][0],r=c[x][1];
    if(sz[l]+1==k)return x;
    if(sz[l]>=k)return findk(l,k);
    return findk(r,k-sz[l]-1);
}
int n,rt;
int ans[NN];

struct node{
    int val,idx;
    bool operator < (const node x) const{
        if(val!=x.val) return val<x.val;
        return idx<x.idx;
    }
}b[NN];


void dfs(int x){
    if(fa[x]!=0)dfs(fa[x]);
    push(x);
}


int main(){
    while(scanf("%d",&n)!=EOF){
        if(n==0)break;
        a[1]=a[n+2]=0;
        rep(i,1,n)scanff(a[i+1]),b[i].idx=i+1,b[i].val=a[i+1];
        rep(i,1,n+2)c[i][0]=c[i][1]=0,rev[i]=0;
        built(1,n+2,0);
        rt=(n+3)>>1;
        sort(b+1,b+1+n);
        rep(i,1,n){
            dfs(b[i].idx);
            splay(b[i].idx,rt);
            ans[i]=sz[c[b[i].idx][0]];
            int x=findk(rt,i);
            int y=findk(rt,ans[i]+2);
            splay(x,rt);
            splay(y,c[x][1]);
            rev[c[y][0]]=1;
        }
        printf("%d",ans[1]);
        rep(i,2,n)printf(" %d",ans[i]);
        printf("\n");
    }

    return 0;
}