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