[HDU5592] ZYB's Premutation

[HDU5592] ZYB's Premutation

题目大意:一个由\([1,n]\)组成的数列，但不知道具体排列，但给出每个前缀的逆序对数目，让你还原排列

Solution

创造一颗\([1,n]\)的权值线段树，初始权值都为\(1\),我们从后往前离线处理，每次拿到一个前缀的逆序对数\(p[i]\)，说明在\(i\)上的这个数字\(a_i\)前面有\(sum=p[i]-p[i-1]\)个数字比它大，所以\(a_i\)是\(a_1,\cdots ,a_i\)中第\(i -sum\)大的数字，查询后这个\(a_i\)对前面就没有用了，需要在权值线段树上删去

Code

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define sc(x) scanf("%d", &x)
#define pf(x) printf("%d\n",x)
#define lson cur << 1
#define rson (cur << 1) | 1

const int N = 50005;

int num[N << 2], val[N << 2];
int p[N], ans[N];
int n;

void build(int cur, int l, int r){
    num[cur] = r - l + 1;
    if(l == r){
        val[cur] = l;
        return;
    }else{
        int mid = l + ((r - l) >> 1);
        build(lson, l, mid);
        build(rson, mid + 1, r); 
    }
}

int query(int cur, int l, int r, int k){
    int mid = l + ((r - l) >> 1); 
    if(l == r){
//      pf(val[cur]);
        return val[cur];
    }else{
        if(num[lson] >= k){
            return query(lson, l, mid, k);
        }else{
            return query(rson, mid + 1, r, k - num[lson]);
        }
    } 
}

void update(int cur, int l, int r, int k){
    num[cur]--;
    if(l == r){
        val[cur] = 0;
        return;
    }else{
        int mid = l + ((r - l) >> 1);
        if(k <= mid){
            update(lson, l, mid, k);
        }else{
            update(rson, mid + 1, r, k);
        }
    }
}

int main(){
    int t;
    scanf("%d", &t);
    while(t--){
        scanf("%d", &n);
        memset(num, 0, sizeof(num));
        memset(val, 0, sizeof(val));
        build(1, 1, n + 1);//l和r要和数组对应起来，不能建的时候是n+1,query和update时又变成了n 
        for(int i = 1, la; i <= n; ++i){
            scanf("%d", &p[i]);
        }
        for(int i = n; i; --i){
            ans[i] = query(1, 1, n + 1, i - p[i] + p[i - 1]);
            update(1, 1, n + 1, ans[i]);
        }
        for(int i = 1; i < n; ++i){
            printf("%d ", ans[i]);
        }
        printf("%d\n", ans[n]);
    }
    return 0;
}

Error

• \(l\)和\(r\)要和数组对应起来，不能建的时候是\(n+1\),\(query\)和\(update\)时又变成了\(n\)

转载于:https://www.cnblogs.com/LMSH7/p/9578098.html