链接:点击打开链接
题意:给出数组a,a[i]表示1~i-1的逆序数对的个数,输出原始序列
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int siz=50005;
int a[siz],sum[siz<<2];
void build(int l,int r,int rt){
if(l==r){
sum[rt]=1;
return;
}
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void Change(int p,int l,int r,int rt){
if(l==r){
sum[rt]=0;
return;
}
int m=(l+r)>>1;
if(p<=m)
Change(p,l,m,rt<<1);
else
Change(p,m+1,r,rt<<1|1);
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
int query(int L,int R,int l,int r,int rt){
if(L<=l&&r<=R)
return sum[rt];
int m=(l+r)>>1,ans=0;
if(L<=m)
ans+=query(L,R,l,m,rt<<1);
if(R>m)
ans+=query(L,R,m+1,r,rt<<1|1);
return ans;
}
int main(){
int t,n,i,j,l,r,ans,mid,tmp;
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,n,1); //每次a[i]增加的值就是之前比
for(i=n;i>=1;i--){ //第i个数大的数的个数,因此相当于
tmp=i-(a[i]-a[i-1]); //求第i个数是区间中没用过的数的第
l=1,r=n; //tmp小,直接二分前缀和,找第一个
while(l<=r){ //前缀和是tmp的数
mid=(l+r)>>1;
if(query(1,mid,1,n,1)>=tmp){
ans=mid;
r=mid-1;
}
else
l=mid+1;
}
a[i]=ans;
Change(ans,1,n,1);
}
for(i=1;i<=n;i++){
if(i!=1)
printf(" ");
printf("%d",a[i]);
}
printf("\n");
}
return 0;
}