spoj227 树状数组插队序列问题

插队问题和线段树解决的方式一样，每个结点维护值的信息是该节点之前的空位有多少，然后从后往前插点即可

注意该题要求输出的是从左往右输出每个士兵的等级，即问士兵最后排在第几个位置

/*
树状数组维护前i个位置的空位
倒序插入，每次二分找到最右边的空位
*/
#include<bits/stdc++.h>
using namespace std;
#define maxn 200005
#define ll long long
ll bit[maxn];
int n,num[maxn],res[maxn];
void add(int x,int num){
    for(int i=x;i<=n;i+=i&-i)
        bit[i]+=num;
}
ll query(int x){
    ll res=;
    for(int i=x;i;i-=i&-i)
        res+=bit[i];
    return res;
}
int vis[maxn];
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        memset(bit,,sizeof bit);
        memset(res,,sizeof res);
        scanf("%d",&n);
        for(int i=;i<=n;i++){
            scanf("%d",&num[i]);
            add(i,);
        }
        for(int i=n;i>=;i--){
            int pos=i-num[i];
            int l=,r=n,ans=;
            while(l<=r){
                int mid=l+r>>;
                if(query(mid)>=pos){//左边的空位小于序号
                    ans=mid;
                    r=mid-;
                }
                else l=mid+;
            }
            res[i]=ans;//注意最后问的是从左到右每个士兵的等级！
            add(ans,-);
        }
        for(int i=; i<=n; ++i)
            if(i==) printf("%d",res[i]);
            else printf(" %d",res[i]);
        printf("\n");
    }
}