hdu5592(线段树+二分)

链接：点击打开链接

题意：给出数组a，a[i]表示1~i-1的逆序数对的个数，输出原始序列

代码：

#include
#include
#include
#include
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;
}