权值线段树求第k小。
题意给n个数所有前缀中的逆序对数,构造出原序列。
第i个前缀减去第i-1个前缀就是第i个值插入序列后增加的逆序对,也就是说前面有这么多数比当前数大。那么i减去这个数代表当前数是前i个数的区间第k小。
维护一个权值线段树,初始节点权值分别为1...n。
从后向前考虑,每遇到一个数就是求当前权值线段树的区间第k小,再把这个数从线段树中删除即可。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
![](https://images.cnblogs.com/OutliningIndicators/ExpandedBlockStart.gif)
#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; }