AcWing 244. 谜一样的牛

有 n 头奶牛，已知它们的身高为 1∼n且各不相同，但不知道每头奶牛的具体身高。

现在这 n头奶牛站成一列，已知第 i 头牛前面有 Ai 头牛比它低，求每头奶牛的身高。

输入格式

第 1 行：输入整数 n。

第 2..n 行：每行输入一个整数 Ai，第 i行表示第 i 头牛前面有 Ai 头牛比它低。
（注意：因为第 1 头牛前面没有牛，所以并没有将它列出）

输出格式

输出包含 n 行，每行输出一个整数表示牛的身高。

第 i 行输出第 i 头牛的身高。

数据范围

1≤n≤1e5

输入样例：

5
1
2
1
0

输出样例：

2
4
5
3
1

代码实现：

#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+5;
int a[N],b[N],tr[N];
int n;
int lowbit(int x){
    return x&-x;
}
void add(int x,int y){
    for(int i=x;i<=n;i+=lowbit(i))tr[i]+=y;
}
int query(int x){
    int res=0;
    for(int i=x;i;i-=lowbit(i))res+=tr[i];
    return res;
}
int main(){
    cin>>n;
    for(int i=2;i<=n;i++)cin>>a[i];
    for(int i=1;i<=n;i++)add(i,1);
    for(int i=n;i;i--){
        int h=a[i]+1;
        int l=1,r=n;
        while(l<r){
            int mid=l+r>>1;
            if(query(mid)>=h)r=mid;
            else l=mid+1;
        }
        b[i]=l;
        add(l,-1);
    }
    for(int i=1;i<=n;i++)cout<<b[i]<<endl;
    return 0;
}