【splay】BZOJ 1152 && 3506:[cqoi2014]排序机械臂

BZOJ 1152 && 3506:[cqoi2014]排序机械臂

---

Description

---

Input

输入共两行，第一行为一个整数N，N表示物品的个数，第二行为N个用空格隔开的正整数，表示N个物品最初排列的编号。

---

Output

输出共一行，N个用空格隔开的正整数P1,P2,P3…Pn，（1 < = Pi < = N），Pi表示第i次操作前第i小的物品所在的位置。 注意：如果第i次操作前，第i小的物品己经在正确的位置Pi上，我们将区间[Pi,Pi]反转(单个物品)。

---

Sample Input

6
3 4 5 1 6 2

---

Sample Output

4 6 4 5 6 6

---

HINT

1<=N<=100000

---

Solution

普通的序列splay…
每个数对应在splay中的编号不变
维护区间最小值所在的编号
每次rev标记一下即可
初始要离散化

Code

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>

using namespace std;

#define maxn 100233

struct splay{
    int ch[2],fa,size,v;
    bool rev;
}t[maxn*2];

int pos[maxn],mn[maxn],ans[maxn];

inline int in()
{
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-'0',ch=getchar();
    return x;
}

int root;

struct sor{
    int pos,v;
}a[maxn];

bool cmp(const sor A,const sor B){return (A.v<B.v||(A.v==B.v&&A.pos<B.pos));}

bool cmp1(const sor A,const sor B){return A.pos<B.pos;}

void pushdown(int k)
{
    t[k].rev=0,t[t[k].ch[0]].rev^=1;t[t[k].ch[1]].rev^=1;
    swap(t[k].ch[0],t[k].ch[1]);
}

int find(int k,int x)
{
    if(t[k].rev)pushdown(k);
    if(t[t[k].ch[0]].size>=x)return find(t[k].ch[0],x);
    else if(x>t[t[k].ch[0]].size+1)return find(t[k].ch[1],x-t[t[k].ch[0]].size-1);
    else return k;
}

void update(int k)
{
    int l=t[k].ch[0],r=t[k].ch[1];
    t[k].size=t[l].size+t[r].size+1;
    mn[k]=t[k].v,pos[k]=k;
    if(mn[l]<mn[k]){mn[k]=mn[l];pos[k]=pos[l];}
    if(mn[r]<mn[k]){mn[k]=mn[r];pos[k]=pos[r];}
}


void rotate(int x,int &k,int d)
{
    int y,z;
    y=t[x].fa;z=t[y].fa;
    if(y==k)k=x;
    else {if(t[z].ch[0]==y)t[z].ch[0]=x;else t[z].ch[1]=x;}
    t[y].fa=x,t[x].fa=z,t[t[x].ch[d]].fa=y;
    t[y].ch[d^1]=t[x].ch[d];t[x].ch[d]=y;
    update(y),update(x);
}

int s[maxn],top=0;

void splay(int x,int &k)
{
    top=0;s[++top]=x;
    for(int i=x;t[i].fa;i=t[i].fa)
        s[++top]=t[i].fa;
    for(;top;top--)
        if(t[s[top]].rev)pushdown(s[top]);
    int y,z;
    while(x!=k)
    {
        y=t[x].fa;z=t[y].fa;
        if(y!=k)
        {
            if(t[z].ch[0]==y&&t[y].ch[0]==x)rotate(y,k,1);
            else if(t[z].ch[1]==y&&t[y].ch[1]==x)rotate(y,k,0);
            else if(t[z].ch[0]==y&&t[y].ch[1]==x)rotate(x,k,0);
            else if(t[z].ch[1]==y&&t[y].ch[0]==x)rotate(x,k,1);
        }
        if(t[t[x].fa].ch[0]==x)rotate(x,k,1);
        else rotate(x,k,0);
    }
}

void build(int l,int r,int f)
{
    if(r<l)return;
    if(l==r)
    {
        mn[l]=a[l].v;
        t[l].size=1;t[l].fa=f;
        t[l].v=a[l].v;
        pos[l]=l;
        t[f].ch[r>f]=r;
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid-1,mid);build(mid+1,r,mid);
    t[mid].v=a[mid].v;
    update(mid);t[mid].fa=f;
    t[f].ch[mid>f]=mid;
}

int querymn(int L,int R)
{
    int ans1=find(root,L),ans2=find(root,R+2);
    splay(ans1,root),splay(ans2,t[root].ch[1]);
    return pos[t[ans2].ch[0]];
}

void rever(int L,int R)
{
    int ans1=find(root,L),ans2=find(root,R+2);
    splay(ans1,root),splay(ans2,t[root].ch[1]);
    t[t[ans2].ch[0]].rev^=1;
}

int main()
{
    freopen("1552.in","r",stdin);
    int n;
    n=in();
    a[1].v=a[n+2].v=99999999,mn[0]=99999999;
    for(int i=2;i<=n+1;i++)
    {
        a[i].v=in();
        a[i].pos=i-1;
    }
    sort(a+2,2+a+n,cmp);
    for(int i=2;i<=n+1;i++)a[i].v=i-1;
    sort(a+2,a+2+n,cmp1);
    build(1,n+2,0);
    root=(n+3)>>1;
    for(int i=1;i<=n;i++)
    {
        int x=querymn(i,n);
        splay(x,root);
        ans[i]=t[t[x].ch[0]].size;
        rever(i,ans[i]);
    }
    for(int i=1;i<=n;i++)printf("%d ",ans[i]);
    return 0;
}