ZYB's Premutation HDU - 5592（线段树的巧妙利用）详解

Problem Description

ZYB has a premutation P,but he only remeber the reverse log of each prefix of the premutation,now he ask you to
restore the premutation.

Pair (i,j)(i<j) is considered as a reverse log if Ai>Aj is matched

 

Input

In the first line there is the number of testcases T.

For each teatcase:

In the first line there is one number N.

In the next line there are N numbers Ai,describe the number of the reverse logs of each prefix,

The input is correct.

1≤T≤5,1≤N≤50000

 

Output

For each testcase,print the ans.

 

Sample Input

1

3

0 1 2

Sample Output

3 1 2

 

这个的大概意思是给你一串数，第i个数表示1到i个数中逆序数，让你还原原数列。

逆序数的定义：在一个排列中，如果一对数的前后位置与大小顺序相反，即前面的数大于后面的数，那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。一个排列中所有逆序总数叫做这个排列的逆序数。也就是说，对于n个不同的元素，先规定各元素之间有一个标准次序（例如n个 不同的自然数，可规定从小到大为标准次序），于是在这n个元素的任一排列中，当某两个元素的先后次序与标准次序不同时，就说有1个逆序。一个排列中所有逆序总数叫做这个排列的逆序数。（百度大法好）

样例解释

3 对应的逆序数为0

3 1对应的逆序数为1

3 1 2对应的逆序数为2

 

下面讲解解法

下面是递归线段树的版本，对线段树不了解的话可以看岩之痕大佬的博客，下面是链接

https://blog.csdn.net/zearot/article/details/48299459

 

以样例为例，用Sum数组记录，如下图

 

 

下面按输入反着遍历

输入

0 1 2

原序列

3 1 2

用arr[i]记录输入(从下标1开始记录），则arr[1] = 0, arr[2] = 1, arr[3] = 2;

m = arr[3] - arr[2], m表示第三个数所能形成逆序对的个数， 2只能与3形成逆序对所以为1

用c表示剩余的数的个数

k = c - m, k表示该数为剩余的数中是第K大的数，因为是反向遍历，m也代表第i个数前面比第i个数小的数的个数。

所以样例中2为剩余数中第二大的数。

下面讲解如和利用线段树判断第k大的数

首先是建树

void Build(int l,int r,int rt){
    if(l==r) {
        Sum[rt] = 1; // 每个数都没呗取走过，所以赋值为1
        return;
    }
    int m=(l+r)>>1;
    Build(l,m,rt<<1);
    Build(m+1,r,rt<<1|1);
    //更新信息 
    PushUp(rt);
}

从节点1开始向下查询，若左儿子的值大于k，则向左儿子移动，否则，则将 k -= (左儿子的值）后，向右儿子移动。下面为代码，也是本题的核心代码。此处建议手动模拟一下样例，就会很容易理解该代码。

int Query(int l, int r, int rt, int k){
    if(l == r){
        return l; //重点 ！！！返回的l为剩余的数中第k大的数对应的值
    }

    int m = (l + r) >> 1;

    if(Sum[(rt << 1)] >= k){ //
        return Query(l, m, rt << 1, k); //如果左儿子的值大于k，那么向左儿子查询
    } else{
        return Query(m + 1, r, rt << 1 | 1, k - Sum[rt << 1]);
    }
}

得到值后将该点的值变为0

ans[i] = Query(1, n, 1, k);
Update(ans[i], 0, 1, n, 1);

void Update(int L,int C,int l,int r,int rt){
    if(l==r){
        Sum[rt] = C;
        return;
    }
    int m=(l+r)>>1;
    if(L <= m) Update(L,C,l,m,rt<<1);
    else       Update(L,C,m+1,r,rt<<1|1);
    PushUp(rt);
} 

下面是全代码

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#define LL long long
using namespace std;

int INF = 0x3f3f3f3f;
const int MAX = 50000 + 20 ;

int Sum[MAX << 2];

int n; // 存数组下标，[1,n]

int arr[MAX];
int ans[MAX];

void PushUp(int rt){
    Sum[rt] = Sum[rt<<1] + Sum[rt<< 1|1];
}

void Build(int l,int r,int rt){
    if(l==r) {
        Sum[rt] = 1; // 每个数都没呗取走过，所以赋值为1
        return;
    }
    int m=(l+r)>>1;
    Build(l,m,rt<<1);
    Build(m+1,r,rt<<1|1);
    //更新信息 
    PushUp(rt);
}



//点修改 A[L] + C
void Update(int L,int C,int l,int r,int rt){
    if(l==r){
        Sum[rt] = C;
        return;
    }
    int m=(l+r)>>1;
    if(L <= m) Update(L,C,l,m,rt<<1);
    else       Update(L,C,m+1,r,rt<<1|1);
    PushUp(rt);
} 


int Query(int l, int r, int rt, int k){
    if(l == r){
        return l; //重点 ！！！返回的l为剩余的数中第k大的数对应的值
    }

    int m = (l + r) >> 1;

    if(Sum[(rt << 1)] >= k){ //
        return Query(l, m, rt << 1, k); //如果左儿子的值大于k，那么向左儿子查询
    } else{
        return Query(m + 1, r, rt << 1 | 1, k - Sum[rt << 1]);
    }
}

int main(int argc, char const *argv[])
{


    int t;
    scanf("%d", &t);
    while(t--){
        scanf("%d", &n);
        for(int i = 1; i <= n; i++){
            scanf("%d", &arr[i]);
        }
        arr[0] = arr[1];
        Build(1, n, 1);
        int c = n;
        for(int i = n; i > 0; i--){
            int k = c - (arr[i] - arr[i - 1]); // arr[i] - arr[i - 1] 的结果为第i个数对应的逆序数, k表示该数为剩余的数中第k大的数
            ans[i] = Query(1, n, 1, k);
            Update(ans[i], 0, 1, n, 1); //取走对应的数后将A[i]变为0
            c--;
        }

        for(int i = 1; i < n; i++){
            printf("%d ", ans[i]);
        }
        printf("%d\n", ans[n]);
    }
    return 0;
}