N头奶牛排队,它们的身高为1~n,知道每头牛前面有多少头比自己矮,求每头牛的身高。

第一头牛前面是0,我们不知道他多高,但最后一头牛前面有n个比自己矮的话就可以知道他的高度是n+1

如果知道最后一头牛的高度是n+1,然后知道倒数第二头牛前面有m个比自己矮,那么就知道了倒数第二头牛的高度为m+1(m>=n),或者m(m<n)

其实就是动态的维护前缀和,并且单点查询

#include <bits/stdc++.h>
#define form(i,n) for(int i=1;i<=n;i++)
#define forn(i,n) for(int i=0;i<n;i++)
#define mst(a) memset(a,0,sizeof(a))
using namespace std;
typedef pair<int,int> P ;
const int N=1e5+90;
typedef unsigned long long ull;
typedef long long ll;
const int MAX_N=4e5+20;
ll a[N],b[N],c[N],n;
ll lowbit(ll x){
    return x&-x;
}
void add(ll x,ll y){
    while (x<=n){
        b[x]+=y;
        x+=lowbit(x);
    }
}
ll ask(ll x){
    ll ans=0;
    while (x){
        ans+=b[x];
        x-=lowbit(x);
    }
    return ans;
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++)b[i]=lowbit(i);
    for(int i=2;i<=n;i++){
        scanf("%d",&c[i]);
    }
    for(int i=n;i;i--){
        int k=c[i]+1;
        ll l=1,r=n;
        while (l<r){
            ll mid=l+r>>1;
            if(ask(mid)>=k)
                r=mid;
            else
                l=mid+1;
        }
        a[i]=l;
        add(l,-1);
    }
    for(int i=1;i<=n;i++)printf("%d\n",a[i]);
}