HDU 5592 线段树

HDU 5592
题目链接：
http://acm.hdu.edu.cn/showproblem.php?pid=5592
题意：
一串1-n的排列。(n <= 5000)
定义a[i]为前i个数的逆序对数，然后要还原这个排列。
思路：
现场的时候有思路，但是线段树不会敲就放弃了。
简单说一下思路，从后往前遍历，由a[i] - a[i - 1]知道当前这个位置放的是剩下的数中第几大的数。

着重说一下线段树怎么写。
设点数n = 5.
一开始把n个数置为1
1 1 1 1 1
线段树里面存的是前缀和
1 2 3 4 5
然后每次根据前缀和查询出刚好为这个值的位置，类似二叉查找树。
if(u > query(o * 2) return query(o * 2 + 1)
Else return query(o * 2)
更新的时候把这个需要更新的叶子节点置为0，然后向上更新前缀和。

看似简单，实则用的很活。平时查找和更新都是要找结点的标记来查找，现在通过值来查找标记。学到了~
源码：

#include <cstdio>#include <cstring>#include <cstdlib>#include <cmath>#include <algorithm>#include <iostream>#include <string>#include <vector>#include <iterator>using namespace std;const int MAXN = 50000 + 5;#define LL long longLL a[MAXN];int ans[MAXN];struct Tree{    int l, r;    int val;}tree[MAXN * 4];void push_up(int o){    tree[o].val = tree[o * 2 + 1].val + tree[o * 2].val;}void build(int l, int r, int o){    tree[o].l = l, tree[o].r = r, tree[o].val = 0;    if(l == r){        tree[o].val = 1;        return;    }    int mid = (l + r) / 2;    build(l, mid, o * 2);    build(mid + 1, r, o * 2 + 1);    push_up(o);}void update(int u, int val, int o){//    printf("u = %d, val = %d, tree[o].l = %d, tree[o].r = %d\n", u, val, tree[o].l, tree[o].r);    if(tree[o].l == tree[o].r){        tree[o].val += val;        return;    }    int mid = (tree[o].l + tree[o].r) >> 1;    if(u <= mid)    update(u, val, o * 2);    else    update(u, val, o * 2 + 1);    push_up(o);}int query(int u, int o){    if(tree[o].l == tree[o].r)  return tree[o].l;    if(u > tree[o * 2].val)   return query(u - tree[o * 2].val, o * 2 + 1);    else    return query(u, o * 2);}int main(){    int T;    scanf("%d", &T);    while(T--){        int n;        scanf("%d", &n);        build(1, n, 1);//        printf("tree[1].l = %d, tree[1].r = %d\n", tree[1].l, tree[1].r);        for(int i = 1 ; i <= n ; i++)   scanf("%I64d", &a[i]);        a[0] = 0;        for(int i = n ; i >= 1 ; i--){            int pian = i- (a[i] - a[i - 1]);//            printf("pian = %d\n", pian);            ans[i] = query(pian, 1);//            printf("ans[i] = %d\n", ans[i]);            update(ans[i], -1, 1);        }        for(int i = 1 ; i <= n ; i++){            printf("%d", ans[i]);            if(i == n)  printf("\n");            else    printf(" ");        }    }    return 0;}