【bzoj3506/1552】【robotic sort】【splay】

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

题解：splay维护区间最小值以及最小值的位置。

因为有相等的权值所以一开始排一遍序即可。

代码：

#include<iostream>
#include<cstdio>
#include<cstring> 
#include<algorithm> 
#define N 100010
#define inf 2100000000
using namespace std;
int n,rev[N],c[N][3],d[N],sz,rt,v[N],size[N],fa[N],mn[N],pos[N],ans[N],s[N];
struct use{int v,p;}a[N];
bool cmp(use a,use b){return a.v<b.v||(a.v==b.v&&a.p<b.p);}
bool cmp2(use a,use b){return a.p<b.p;}
void update(int k){
  int a=c[k][0],b=c[k][1];
  mn[k]=v[k];pos[k]=k;
  if (mn[a]<mn[k]) mn[k]=mn[a],pos[k]=pos[a];
  if (mn[b]<mn[k]) mn[k]=mn[b],pos[k]=pos[b];  
  size[k]=size[a]+size[b]+1;
}
void pushdown(int k){
  int a=c[k][0],b=c[k][1];
   rev[k]^=1;rev[a]^=1;rev[b]^=1;
   swap(c[k][0],c[k][1]);
}
void rotata(int x,int &k){
   int y=fa[x],z=fa[y],l,r;
   if (c[y][0]==x) l=0;else l=1;r=l^1;
   if (y!=k){if (c[z][0]==y) c[z][0]=x;else c[z][1]=x;}
   else k=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){
  int top=0;s[++top]=x;
  for(int i=x;fa[i];i=fa[i]) s[++top]=fa[i];
  for(int i=top;i;i--) if(rev[s[i]])pushdown(s[i]);
  while (x!=k){
    int y=fa[x],z=fa[y];
    if (y!=k){
      if (c[y][0]==x^c[z][0]==y) rotata(x,k);
      else rotata(y,k);
    }
    rotata(x,k);
  }
}
void build(int l,int r,int f){
  if (l>r) return;
  if (l==r){
   int now=l,pre=f;fa[now]=pre;
   if (l<f) c[pre][0]=now;
   else c[pre][1]=now;
   v[now]=mn[now]=a[l].v;pos[now]=l;size[now]=1;
   return;
  }	
  int mid=(l+r)>>1,now=mid,pre=f;
  build(l,mid-1,mid);build(mid+1,r,mid);
  fa[now]=pre;v[now]=mn[now]=a[now].v;
  pos[now]=now;update(now);
  if (mid<f) c[pre][0]=now;
  else c[pre][1]=now;
}
int find(int k,int rank){
  if (rev[k]) pushdown(k);
  int l=c[k][0],r=c[k][1];
  if(size[l]+1==rank) return k;
  else if (size[l]>=rank) return find(l,rank);
  else return find(r,rank-size[l]-1); 
}
int query(int l,int r){
  int x=find(rt,l),y=find(rt,r+2);
  //cout<<x<<' '<<y<<endl;
  splay(x,rt);splay(y,c[x][1]);
  int z=c[y][0];
  return pos[z];
}
void rever(int l,int r){
  int x=find(rt,l),y=find(rt,r+2);
  splay(x,rt);splay(y,c[x][1]);
  int z=c[y][0];
  rev[z]^=1;
}
int main(){
  scanf("%d",&n);a[1].v=inf;a[n+2].v=inf;mn[0]=inf;
  for (int i=2;i<=n+1;i++) scanf("%d",&a[i].v),a[i].p=i;
  sort(a+2,a+n+2,cmp);
  for (int i=2;i<=n+1;i++) a[i].v=i-1;
  sort(a+2,a+n+2,cmp2);
  build(1,n+2,0);rt=(n+3)>>1;
  //for (int i=1;i<=n+2;i++) cout<<size[i]<<endl;
  for (int i=1;i<=n;i++){
    int x=query(i,n);
	splay(x,rt);
	ans[i]=size[c[x][0]];
	rever(i,ans[i]);
  }
  for (int i=1;i<=n;i++){
    printf("%d",ans[i]);
    if (i!=n) printf(" ");
  }
}