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=0;
    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,0,sizeof bit);
        memset(res,0,sizeof res);
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d",&num[i]);
            add(i,1);
        }
        for(int i=n;i>=1;i--){
            int pos=i-num[i];
            int l=1,r=n,ans=1;
            while(l<=r){
                int mid=l+r>>1;
                if(query(mid)>=pos){//左边的空位小于序号 
                    ans=mid;
                    r=mid-1;
                }
                else l=mid+1;
            }
            res[i]=ans;//注意最后问的是从左到右每个士兵的等级！ 
            add(ans,-1);
        }
        for(int i=1; i<=n; ++i)
            if(i==1) printf("%d",res[i]);
            else printf(" %d",res[i]);
        printf("\n");
    }
}