SPOJ

题目链接

题意：

原题给的背景太繁琐，直接简化一下；

给定一个序列 $A$ , 其中 $A[i]$ 表示 $i$ 之前有多少个比当前数大，现在求原序列，序列值为 $1...n$；

思路：

倒序还原，查询当前未使用的数字中的 第 $k$ 大 是多少，用树状数组维护当前值是否被使用，二分查询 第 $k$ 大 ，复杂度 $O( n logn log n)$。

/*
*  Author: windystreet
*  Date  : 2018-08-11 12:50:35
*  Motto : Think twice, code once.
*/
#include<bits/stdc++.h>

using namespace std;

#define X first
#define Y second
#define eps  1e-5
#define gcd __gcd
#define pb push_back
#define PI acos(-1.0)
#define lowbit(x) (x)&(-x)
#define bug printf("!!!!!
");
#define mem(x,y) memset(x,y,sizeof(x))

typedef long long LL;
typedef long double LD;
typedef pair<int,int> pii;
typedef unsigned long long uLL;

const int maxn = 2e5+7;
const int INF  = 1<<30;
const int mod  = 1e9+7;
struct node
{
    int v,id;
}s[maxn];
int tree[maxn];
int vis[maxn];
int n;
void init(){
    for(int i=0;i<=n;i++)
        tree[i] = 0,vis[i] = 0;
    return; 
}

void add(int x,int v){
    for(int i=x;i<=n;i+=lowbit(i))
        tree[i] += v;
    return ;
}

int query(int x ){
    int res = 0;
    for(int i=x ;i;i-=lowbit(i))
        res += tree[i];
    return res;
}

void solve(){
    scanf("%d",&n);
    init();
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i].v);
        add(i,1);                                  // 将每一个数字表示为存在 
    }
    for(int i=n;i>=1;i--){
        int x = i - s[i].v;                        //  k
        int L = 1,R = n;
        while(L <= R){                             // 二分查找
            int mid = (L+R)>>1;                    // 若当前的值大于等于k则记录答案
            if(query(mid)>=x)s[i].id = mid,R = mid-1;
            else L = mid + 1;
        }
        add(s[i].id,-1);                           // 使用了之后就取消标记
    }
    for(int i=1;i<=n;i++){
        printf("%d%c",s[i].id," 
"[i==n]);
    }
    return;
}

int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
//    ios::sync_with_stdio(false);
    int t = 1;
    scanf("%d",&t);
    while(t--){
    //    printf("Case %d: ",cas++);
        solve();
    }
    return 0;
}