Bestcoder round #65 && hdu 5592 ZYB's Premutation 线段树

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <iostream>
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1|1
using namespace std;
const int INF = 0x3f3f3f3f;
typedef long long ll;

const int N = 50005;
int num[N << 2], a[N], ans[N];
void up(int rt) { num[rt] = num[rt << 1] + num[rt << 1|1]; }
void build(int l, int r, int rt)
{
    if(l == r) {
        num[rt] = 1;
        return ;
    }
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    up(rt);
}
void update(int l, int r, int rt, int p)
{
    if(l == p && r == p) {
        num[rt]--;
        return ;
    }
    int m = (l + r) >> 1;
    if(p <= m) update(lson, p);
    else update(rson, p);
    up(rt);
}
int pos;
void get(int l, int r, int rt, int x)
{
    if(l == r) {
        pos = l;
        return;
    }
    int m = (l + r) >> 1;
    if(num[rt << 1] < x) get(rson, x - num[rt << 1]);
    else get(lson, x);
}
int main()
{
    int _; scanf("%d", &_);
    while(_ --)
    {
        int n; scanf("%d", &n);
        for(int i = 1; i <= n; ++i) scanf("%d", &a[i]);
        build(1, n, 1);
        int c = n;
        for(int i = n; i >= 2; --i)
        {
            int d = c - (a[i] - a[i - 1]);
            get(1, n, 1, d);
            update(1, n, 1, pos);
         //   cout << pos << endl;
            ans[i] = pos;
            c--;
        }
        get(1, n, 1, 1); ans[1] = pos;
        for(int i = 1; i < n; ++i) printf("%d ", ans[i]);
        printf("%d\n", ans[n]);
    }
    return 0;
}