HDU 1890 splay

点击打开链接

题意：将n个数排序，排序要遵循的是反转操作，每次输出需要反转的位置

思路：因为要反转那么splay就可以了，然后注意的是节点对应序列的位置，然后每个数的实际位置记录一下，每次将这个数旋转到根节点，然后左儿子的size+当前旋转的数就是结果，然后将根节点删除

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=100010;
int pre[maxn],ch[maxn][2],size[maxn],root,tot1;//父节点，左右孩子，子树规模，根节点，节点数量
int flag[maxn],rev[maxn],m[maxn];//该点的值，懒惰标记
//int a[maxn];
int n,q,key[maxn];
int s[maxn],tot2;
char Istr[maxn];
void treaval(int x){//debug部分
    if(x){
        treaval(ch[x][0]);
        printf("结点%2d:左儿子 %2d 右儿子 %2d 父结点 %2d size=%2d,key=%2d add=%2d\n",x,ch[x][0],ch[x][1],pre[x],size[x],key[x],flag[x]);
        treaval(ch[x][1]);
    }
}
void debug(){
    printf("root:%d\n",root);treaval(root);
}
void newnode(int &r,int fa,int k){
    if(tot2) r=s[tot2--];
    else r=k;
    pre[r]=fa;size[r]=1;flag[r]=0;rev[r]=0;
//    key[r]=m[r]=k;
    ch[r][0]=ch[r][1]=0;
}
void update_rev(int r){
    if(r==0) return ;
    swap(ch[r][0],ch[r][1]);
    rev[r]^=1;
}
void update_add(int r,int val){
    if(r==0) return ;
    flag[r]+=val;
    key[r]+=val;
    m[r]+=val;
}
void pushup(int r){
    size[r]=size[ch[r][0]]+size[ch[r][1]]+1;
    m[r]=key[r];
    if(ch[r][0]) m[r]=min(m[r],m[ch[r][0]]);
    if(ch[r][1]) m[r]=min(m[r],m[ch[r][1]]);
}
void pushdown(int r){
    if(rev[r]){
        update_rev(ch[r][0]);
        update_rev(ch[r][1]);
        rev[r]=0;
    }
    if(flag[r]){
        update_add(ch[r][0],flag[r]);
        update_add(ch[r][1],flag[r]);
        flag[r]=0;
    }
}
void buildtree(int &x,int l,int r,int fa){
    if(l>r) return ;
    int mid=(l+r)>>1;
    newnode(x,fa,mid);
    buildtree(ch[x][0],l,mid-1,x);
    buildtree(ch[x][1],mid+1,r,x);
    pushup(x);
}
void init(){
    root=tot1=tot2=0;
    ch[root][0]=ch[root][1]=size[root]=flag[root]=0,rev[root]=0;
//    key[root]=0;m[root]=inf;
//    newnode(root,0,inf);
//    newnode(ch[root][1],root,inf);
    buildtree(root,1,n,0);
//    pushup(ch[root][1]);pushup(root);
}
void Rotate(int x,int kind){
    int y=pre[x];
    pushdown(y);pushdown(x);
    ch[y][!kind]=ch[x][kind];
    pre[ch[x][kind]]=y;
    if(pre[y]) ch[pre[y]][ch[pre[y]][1]==y]=x;
    pre[x]=pre[y];
    ch[x][kind]=y;pre[y]=x;
    pushup(y);
}
void splay(int r,int goal){
    pushdown(r);
    while(pre[r]!=goal){
        if(pre[pre[r]]==goal){
            pushdown(pre[r]);pushdown(r);
            Rotate(r,ch[pre[r]][0]==r);
        }else{
            pushdown(pre[pre[r]]);pushdown(pre[r]);
            pushdown(r);
            int y=pre[r];
            int kind=ch[pre[y]][0]==y;
            if(ch[y][kind]==r) Rotate(r,!kind),Rotate(r,kind);
            else Rotate(y,kind),Rotate(r,kind);
        }
    }
    pushup(r);
    if(goal==0) root=r;
}
int get_kth(int r,int k){
    pushdown(r);
    int t=size[ch[r][0]]+1;
    if(t==k) return r;
    if(t>k) return get_kth(ch[r][0],k);
    else return get_kth(ch[r][1],k-t);
}
int get_min(int r){
    pushdown(r);
    while(ch[r][0]){
        r=ch[r][0];
        pushdown(r);
    }
    return r;
}
int get_max(int r){
    pushdown(r);
    while(ch[r][1]){
        r=ch[r][1];
        pushdown(r);
    }
    return r;
}
void add(int l,int r,int val){
    splay(get_kth(root,l),0);
    splay(get_kth(root,r+2),root);
    update_add(ch[ch[root][1]][0],val);
    pushup(ch[root][1]);
    pushup(root);
}
void Reverse(int l,int r){
    splay(get_kth(root,l),0);
    splay(get_kth(root,r+2),root);
    update_rev(ch[ch[root][1]][0]);
    pushup(ch[root][1]);pushup(root);
}
//void Insert(int len){
//    splay(get_kth(root,pos+1),0);
//    splay(get_min(ch[root][1]),root);
//    buildtree(ch[ch[root][1]][0],0,len-1,ch[root][1]);
//    pushup(ch[root][1]);pushup(root);
//}
//void erase(int r){
//    if(r){
//        s[++tot2]=r;
//        erase(ch[r][0]);
//        erase(ch[r][1]);
//    }
//}
//void Delete(int x){
//    splay(get_kth(root,pos+1),0);
//    splay(get_kth(root,pos+x+2),root);
//    erase(ch[ch[root][1]][0]);pre[ch[ch[root][1]][0]]=0;
//    ch[ch[root][1]][0]=0;
//    pushup(ch[root][1]);pushup(root);
//}
int query_min(int l,int r){
    splay(get_kth(root,l),0);
    splay(get_kth(root,r+2),root);
    return m[ch[ch[root][1]][0]];
}
void pri(int x){
    if(x==0) return ;
    pushdown(x);
    if(ch[x][0]) pri(ch[x][0]);
    printf("%d\n",key[x]);
    if(ch[x][1]) pri(ch[x][1]);
}
void Revolve(int l,int r){
    splay(get_kth(root,l),0);
    splay(get_kth(root,r+2),root);
    int tmp=ch[ch[root][1]][0];
    ch[ch[root][1]][0]=0;
    pushup(ch[root][1]);pushup(root);
    int len=r-l+1;
    len=n-len+1;
    splay(get_kth(root,len),0);
    splay(get_kth(root,len+1),root);
    ch[ch[root][1]][0]=tmp;
    pre[ch[ch[root][1]][0]]=ch[root][1];
    pushup(ch[root][1]);pushup(root);
}
void Remove(){
    if(ch[root][0]==0){
        root=ch[root][1];
        pre[root]=0;
    }else{
        int mm=get_max(ch[root][0]);
        splay(mm,root);
        ch[mm][1]=ch[root][1];
        pre[ch[root][1]]=mm;
        root=mm;
        pre[root]=0;
        pushup(root);
    }
}
struct pos{
    int id,val;
}a[maxn];
bool cmp(const pos &b,const pos &c){
    if(b.val==c.val) return b.id<c.id;
    return b.val<c.val;
}
int main(){
    int x,y;
    char op[20];
    while(scanf("%d",&n)!=-1){
        if(n==0) break;
        init();
//        debug();
        for(int i=1;i<=n;i++) scanf("%d",&a[i].val),a[i].id=i;
        sort(a+1,a+1+n,cmp);
        for(int i=1;i<n;i++){
            splay(a[i].id,0);
            update_rev(ch[root][0]);
            printf("%d ",i+size[ch[root][0]]);
//            debug();
//            pri(root);
            Remove();
        }
        printf("%d\n",n);
    }
    return 0;
}