线段树 csu1555 Inversion Sequence

题意：告诉你逆序对，还原逆序对

思路：首先我们来手动模拟一下样例

5
1 2 0 1 0

1的前面有1个比1大的，2的前面有2个比2大的，4的前面有一个比4大的

首先，1的位置可以直接确定，因为除了1以外所有的数都比1大，所以1肯定在第二个位置

得到 _ 1 _ _ _

然后考虑2，2的前面有两个数字比2大，因为我们已经确定的数字只有1，1已经比2小了，所以要无视1，其实就是说，2的前面要留下2个空位，这样那2个空位只能放3，4，5，就能保证2的前面一定会有2个数字比它大了，所以

得到 _ 1 _ 2 _

再考虑3，3的前面没有比3大的，1和2都比3小应该直接无视，那么3只能放在第一个位置

得到3 1 _ 2 _

再考虑4，4的前面有一个比它大，所以4的前面应该留一个空格

得到3 1 _ 2 4

最后

得到3 1 5 2 4

这下我们就能找到规律了，从1开始考虑到n，如果第i个数字的前面有A[i]个数比它大，那么第i个数字就应该放在上一个状态剩下的空位中的第A[i]+1的空位位置

然后问题就变成找空位位置了，可以用线段树去维护

如果左区间的空位数足够，就访问左树，否则，就在右树访问x-左区间的空位数

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<cstdio>
#include<string>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
 
using namespace std;
typedef long long LL;
typedef pair<LL, int> PLI;
 
const int MX = 2e4 + 5;
const int INF = 0x3f3f3f3f;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define root 1,n,1
#define RR rt<<1|1
#define LL rt<<1
 
int S[MX << 2], A[MX], B[MX];
 
void push_up(int rt) {
    S[rt] = S[LL] + S[RR];
}
 
void build(int l, int r, int rt) {
    if(l == r) {
        S[rt] = 1;
        return;
    }
 
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    push_up(rt);
}
 
int query(int x, int l, int r, int rt) {
    if(l == r) {
        S[rt] = 0;
        return l;
    }
 
    int m = (l + r) >> 1, ret;
    if(S[LL] >= x) ret = query(x, lson);
    else ret = query(x - S[LL], rson);
    push_up(rt);
    return ret;
}
 
int main() {
    int n;
    //freopen("input.txt","r",stdin);
    while(~scanf("%d", &n)) {
        build(root);
        for(int i = 1; i <= n; i++) {
            scanf("%d", &A[i]);
        }
 
        bool sign = true;
        for(int i = 1; i <= n; i++) {
            if(S[1] < A[i] + 1) {
                sign = false;
                break;
            }
            B[query(A[i] + 1, root)] = i;
        }
 
        if(!sign) printf("No solution\n");
        else {
            for(int i = 1; i <= n; i++) {
                printf("%d%c", B[i], i == n ? '\n' : ' ');
            }
        }
    }
    return 0;
}