uva 11525 Permutation 线段树

题目大意：给定n和k，输出1-k的所有置换按照字典序排列后的第n个置换，其中n由k个数Si给定

当i等于1的时候(k - 1)!就是后面k-1个数所有的置换的个数，然后前面乘si，所以第n个置换的第一个数字即si + 1，同理，可发现其实就是每次求第k大，然后删去该数字。

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 5e4 + 10;

#define lson o * 2, l, m
#define rson o * 2 + 1, m + 1, r
#define ls o * 2
#define rs o * 2 + 1

int sum[N * 4], a[N];

void pushup(int o) {
    sum[o] = sum[ls] + sum[rs];
}

void build(int o, int l, int r) {
    if (l == r) {
        sum[o] = 1;
        return ;
    }
    int m = (l + r) / 2;
    build(lson);
    build(rson);
    pushup(o);
}

void update(int o, int l, int r, int p) {
    if (l == r) {
        sum[o] -= 1;
        return;
    }
    int m = (l + r) / 2;
    if (p <= m) update(lson, p);
    else update(rson, p);
    pushup(o);
}

int query(int o, int l, int r, int k) {
    if (l == r) return l;
    int m = (l + r) / 2;
    if (sum[ls] >= k) return query(lson, k);
    else return query(rson, k - sum[ls]);
}

int main() {
    int t, n;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
        build(1, 1, n);
        for (int i = 1; i < n; i++) {
            int x = query(1, 1, n, a[i] + 1);
            printf("%d ", x);
            update(1, 1, n, x);
        }
        printf("%d\n", query(1, 1, n, a[n] + 1));
    }
    return 0;
}