【CQOI2014】【BZOJ 3506】【JZOJ 3599】排序机械臂

Description

对于100%的数据 1<=n<=100000 ,1<=ai<=2,000,000,000

Analysis

继续用来练splay
首先排个序，求出每次找到最小的数是哪个，当然输出答案要记录size
然后可以模拟操作，关键就是处理翻转操作。
具体过程就是弄一个翻转的lazy tag，做到一个点就执行down：如果tag=1就交换其左右子树，并且下传标记。
为了方便，把当前最小的数删除。

Code

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=100010;
int n,top,t[N][2],f[N],key[N],a[N],size[N],sta[N];
bool bz[N];
bool cmp(int x,int y)
{
    return key[x]<key[y] || key[x]==key[y] && x<y;
}
bool pd(int x)
{
    return x==t[f[x]][1];
}
void update(int x)
{
    size[x]=size[t[x][0]]+size[t[x][1]]+1;
}
void down(int x)
{
    if(!bz[x]) return;
    swap(t[x][0],t[x][1]);
    if(t[x][0]) bz[t[x][0]]^=1;
    if(t[x][1]) bz[t[x][1]]^=1;
    bz[x]=0;
}
void up(int x,int y)
{
    for(;;x=f[x])
    {
        sta[++top]=x;
        if(x==y) break;
    }
    while(top) down(sta[top--]);
}
void rotate(int x)
{
    int y=f[x],z=pd(x);
    t[y][z]=t[x][1-z];
    if(t[x][1-z]) f[t[x][1-z]]=y;
    f[x]=f[y];
    if(f[y]) t[f[y]][pd(y)]=x;
    t[x][1-z]=y,f[y]=x;
    update(y);
}
void splay(int x,int y)
{
    up(x,y);
    while(f[x]!=y)
    {
        if(f[f[x]]!=y)
            if(pd(x)==pd(f[x])) rotate(f[x]);
            else rotate(x);
        rotate(x);
    }
    update(x);
}
void del(int x)
{
    down(x);
    int y=t[x][1];
    while(1)
    {
        down(y);
        if(t[y][0]) y=t[y][0];
        else break;
    }
    splay(y,0);
    f[t[x][0]]=f[x],t[f[x]][0]=t[x][0];
    update(f[x]);
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n)
    {
        scanf("%d",&key[i]),a[i]=i;
        size[i]=i,f[i]=i+1,t[i+1][0]=i;
    }
    size[n+1]=n+1;
    sort(a+1,a+n+1,cmp);
    fo(i,1,n)
    {
        splay(a[i],0);
        printf("%d ",size[t[a[i]][0]]+i);
        bz[t[a[i]][0]]^=1;
        del(a[i]);
    }
    return 0;
}