【BZOJ-1552&3506】robotic sort&排序机械臂 Splay

1552: [Cerc2007]robotic sort

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 806  Solved: 329
[Submit][Status][Discuss]

Description

Input

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

Output

输出共一行，N个用空格隔开的正整数P1,P2,P3…Pn，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

Source

HNOI2009集训Day6

Solution

Splay基本操作...

我们首先对给出的序列排序，第一关键字是编号，第二关键字是时间戳，然后用splay去维护这个有顺序的序列就可以了

或者可以考虑维护min和pos

Splay要常写！调的跟*一样慢

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
inline int read()
{
    int x=0; char ch=getchar();
    while (ch<'0' || ch>'9') {ch=getchar();}
    while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x;
}
#define MAXN 100010
int N,pos[MAXN];
struct Node{int id,t;}a[MAXN];
inline bool cmp(Node A,Node B) {return A.id==B.id? A.t<B.t:A.id<B.id;}
namespace SplayTree
{
    int size[MAXN],root,sz,fa[MAXN],son[MAXN][2]; bool rev[MAXN];
    #define ls(x) son[x][0]
    #define rs(x) son[x][1]
    inline void Update(int now) {size[now]=size[ls(now)]+size[rs(now)]+1;}
    inline void PushDown(int now)
    {
        if (!rev[now]) return;
        rev[ls(now)]^=1; rev[rs(now)]^=1; swap(ls(now),rs(now)); rev[now]=0;
    }
    inline bool Right(int now) {return son[fa[now]][1]==now;}
    inline void rotate(int now)    
    {
        PushDown(fa[now]); PushDown(now);
        int f=fa[now],gf=fa[f],wh=Right(now);
        son[f][wh]=son[now][wh^1]; fa[son[f][wh]]=f;
        fa[f]=now; son[now][wh^1]=f; fa[now]=gf;
        if (gf) son[gf][son[gf][1]==f]=now;
        Update(f); Update(now);
    }
    inline void splay(int now,int tar)
    {
        for (int f; (f=fa[now])!=tar; rotate(now))
            if (fa[f]!=tar) rotate(Right(now)==Right(f)? f:now);
        if (!tar) root=now;    
    }
    inline int GetX(int k)
    {
        int x=root;
        while (1)
            {
                PushDown(x);
                if (k<=size[ls(x)]) x=ls(x);
                    else if (k==size[ls(x)]+1) return x;
                        else k-=size[ls(x)]+1,x=rs(x);
            }
        return -1;
    }
    inline int GetK(int x) {splay(x,0); return size[ls(x)];}
    inline int BuildTree(int l,int r,int last)
    {
        if (r<l) return 0;
        int mid=(l+r)>>1,now=++sz;
        pos[a[mid].id]=now; fa[now]=last;
        int lson=BuildTree(l,mid-1,now),rson=BuildTree(mid+1,r,now);
        son[now][0]=lson; son[now][1]=rson;
        Update(now);
        return now;
    }
    inline void Reverse(int l,int r) {splay(l,0),splay(r,l); rev[ls(rs(root))]^=1;}
    inline void Init() {root=BuildTree(1,N+2,0);}
}
int main()
{
    N=read();
    for (int i=2,t=0; i<=N+1; i++) a[i].id=read(),a[i].t=++t;
    a[1].id=0; a[N+2].id=N+2; 
    stable_sort(a+2,a+N+2,cmp);
    for (int i=1; i<=N; i++) a[a[i+1].t+1].id=i;
//    for (int i=1; i<=N+2; i++) printf("%d ",a[i].id); puts("");
    SplayTree::Init();
    for (int i=1; i<=N; i++)
        {
            int loc=SplayTree::GetK(pos[i]);
            int x=SplayTree::GetX(i),y=SplayTree::GetX(loc+2);
//            printf("%d %d\n",x,y);
            SplayTree::Reverse(x,y);
            printf("%d%c",loc,i!=N? ' ':'\n');
        }
    return 0;
}

和char哥，龙哥，连坐调splay...感觉智障++