csu 1555 Inversion Sequence 线段树反求逆序数

传送门：Inversion Sequence

题目大意

告诉逆序数，求出原逆序数

解题思路

我们可以先看样例怎么推倒出来的：
1 2 0 1 0
1. 在1前面有1个比这个数大的，所以1只能在第二个位置
2. 在2前面有两个比他大的数，其中1又排在了第二个位置，所以2只能排在第四个位置
3. 在3前面有0个比3大的，所以3只能排在第一个位置
4. 在4前面有一个比4大的，所以4只能在最后一个位置
5. 在5前面有0个比5大的，只剩下一个位置了，5排在第三位

通过上面的过程我们能看出，每次输入就是求前面有几个位置没有被占用，如果存在那么下一个位置就是这个数的位置。

AC代码

#include<cstdio>
#include<cstring>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int MAXN = 10000+5;
int sum[MAXN<<2],ans[MAXN];
bool flag;
int pos;
void pushUp(int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt] = 1;
        return ;
    }
    int mid = (l+r)>>1;
    build(lson);
    build(rson);
    pushUp(rt);
}

void query(int l,int r,int rt,int x)
{
    if(sum[rt]<x)
    {
        flag = false;
        return ;
    }
    if(l==r)
    {
        pos = l;
        sum[rt] = 0;
        return ;
    }
    int mid = (l+r)>>1;
    if(x<=sum[rt<<1])
        query(lson,x);
    else
        query(rson,x-sum[rt<<1]);
    pushUp(rt);
}

int main()
{
    int N,x;
    while(scanf("%d",&N)!=EOF)
    {
        build(1,N,1);
        flag = true;
        for(int i=1;i<=N;i++)
        {
            scanf("%d",&x);
            if(!flag) continue;
            query(1,N,1,x+1);
            ans[pos] = i;
        }
        if(!flag)
        {
            puts("No solution");
            continue;
        }
        for(int i=1;i<=N;i++)
        {
            if(i==N)
                printf("%d\n",ans[i]);
            else
                printf("%d ",ans[i]);
        }
    }
    return 0;
}