hdu 5592 ZYB's Premutation

权值线段树求第k小。

题意给n个数所有前缀中的逆序对数，构造出原序列。

第i个前缀减去第i-1个前缀就是第i个值插入序列后增加的逆序对，也就是说前面有这么多数比当前数大。那么i减去这个数代表当前数是前i个数的区间第k小。

维护一个权值线段树，初始节点权值分别为1...n。

从后向前考虑，每遇到一个数就是求当前权值线段树的区间第k小，再把这个数从线段树中删除即可。

 

#include <bits/stdc++.h>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define up rt,rt<<1,rt<<1|1
using namespace std;
const int M = 5e4+7;
int _,n,pos,a[M],ans[M];
struct Tree
{
    int num,val;
}tree[M<<2];
void pushup(int rt,int l,int r){
    tree[rt].num=tree[l].num+tree[r].num;
}
void build(int l,int r,int rt){
    tree[rt].num=0;
    if(l==r){
        tree[rt].num=1;tree[rt].val=l;return ;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(up);
}
void update(int l,int r,int rt){
    if(l==r){
        tree[rt].num--;return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) update(lson);
    else update(rson);
    pushup(up);
}
int query(int l,int r,int rt,int k){
    if(l==r){
        return tree[rt].val;
    }
    int mid=(l+r)>>1;
    if(tree[rt<<1].num>=k) return query(lson,k);
    else return query(rson,k-tree[rt<<1].num);
}
int main(){
    scanf("%d",&_);
    while(_--){
        scanf("%d",&n);
        build(1,n,1);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);a[0]=0;
        for(int i=n;i>=1;i--){
            int k=i-(a[i]-a[i-1]);
            ans[i]=query(1,n,1,k);pos=ans[i];
            update(1,n,1);
        }
        for(int i=1;i<=n;i++) printf("%d%c",ans[i],i==n?'\n':' ');
    }
    return 0;
}