HDU 5592(ZYB's Premutation-线段树)

题意：有一个1到n的排列$P$,已知$P$中每个前缀区间的逆序对数,求这个排列.

写一个线段树，从后往前递推

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<functional>
#include<iostream>
#include<cmath>
#include<cctype>
#include<ctime>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define Forpiter(x) for(int &p=iter[x];p;p=next[p])  
#define Lson (o<<1)
#define Rson ((o<<1)+1)
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,127,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define MEM2(a,i) memset(a,i,sizeof(a));
#define INF (2139062143)
#define F (100000007)
#define MAXN (500000+10)
typedef long long ll;
int read()
{
    int x=0,f=1; char ch=getchar();
    while(!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();}
    while(isdigit(ch)) { x=x*10+ch-'0'; ch=getchar();}
    return x*f;
} 
class SegmentTree  
{  
    ll a[MAXN*4],sumv[MAXN*4],addv[MAXN*4];  
    int n;  
public:  
    SegmentTree(){MEM(a) MEM(sumv) MEM(addv) }  
    SegmentTree(int _n):n(_n){MEM(a) MEM(sumv) MEM(addv) }  
    void mem(int _n)  
    {  
        n=_n;  
        MEMI(a)   
    }  

    void maintain(int o,int L,int R)  
    {
        sumv[o]=0;
        if (L<R) //只考虑左右子树 
        {
            sumv[o]=sumv[Lson]+sumv[Rson];
        } //只考虑add操作 
        sumv[o]+=addv[o]*(R-L+1);
    }

    int y1,y2,v;
    void update(int o,int L,int R)
    {
        if (y1<=L&&R<=y2) {
            addv[o]+=v;
        }
        else{
            int M=(R+L)>>1;
            if (y1<=M) update(Lson,L,M);
            if (M< y2) update(Rson,M+1,R);
        }

        maintain(o,L,R); 
    }
    void build(int o,int L,int R)
    {
        if (L==R) {
            addv[o]=1;  
        }
        else{
            int M=(R+L)>>1;
            build(Lson,L,M);
            build(Rson,M+1,R);
        }
        maintain(o,L,R); 
    }
    int suM_calc(int o,int L,int R,int add) {
        return sumv[o]+add*(R-L+1);
    }
    int query(int o,int L,int R,ll add,int k)
    {
        if (L==R)
        {
            return L;
        } 
        else{
            int M=(R+L)>>1;
            int t1=suM_calc(Lson,L,M,add+addv[o]);
            if (k<=t1 ) 
                return query(Lson,L,M,add+addv[o],k);
            return query(Rson,M+1,R,add+addv[o],k-t1);
        }       
    }
}S;  
int a[100000];
int b[100000];
int main()
{
//  freopen("c.in","r",stdin);
//  freopen(".out","w",stdout);
    int T=read();
    while(T--) {
        int n=read();
        a[0]=0;
        For(i,n) a[i]=read();
        S.mem(n+10);
        S.build(1,1,n);
        ForD(i,n) {
            int p=a[i]-a[i-1]+1;
            p=i-p+1;
            int ans = S.query(1,1,n,0,p);
            b[i]=ans;
            S.y1=S.y2=ans,S.v=-1;
            S.update(1,1,n);
        } 
        For(i,n-1) printf("%d ",b[i]);
        cout<<b[n]<<endl;
    }   
    return 0;
}