谜一样的牛-树状数组

有n头奶牛，已知它们的身高为 1~n 且各不相同，但不知道每头奶牛的具体身高。
现在这n头奶牛站成一列，已知第i头牛前面有AiAi头牛比它低，求每头奶牛的身高。
输入格式
第1行：输入整数n。
第2…n行：每行输入一个整数AiAi,第i行表示第i头牛前面有AiAi头牛比它低。

（注意：因为第1头牛前面没有牛，所以并没有将它列出）
输出格式
输出包含n行，每行输出一个整数表示牛的身高。
第i行输出第i头牛的身高。
数据范围
1≤n≤1051≤n≤105
输入样例：
5
1
2
1
0

输出样例：
2
4
5
3
1

思路：从后往前看每头牛，如果当前看到的牛前面有i头牛比他低的话，那么这头牛就是剩下
的牛中间第i+1高的。俩个问题：如何确定剩下的牛中第i+1高的牛的高度是多少？
当前剩下的牛的高度中（因为牛的高度是1-n不重复）， 前多少高度的牛的数量的和
为i+1的话，那么这个高度就是第轮到的这头牛的高度，这里用二分，然后当这个高度用
过后，这头牛的高度便修改为0

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=100010;
int n;
int ans[N];
int h[N];
int tr[N];
int lowbit(int x){
    return x&-x;
}
void add(int x,int c){
 for(int i=x;i<=n;i+=lowbit(i))    tr[i]+=c;
}
int sum(int x){
 int res=0;
 for(int i=x;i;i-=lowbit(i))      res+=tr[i];
 return res;
}
int main(){
 scanf("%d",&n);
 for(int i=2;i<=n;i++)   scanf("%d",&h[i]);
 for(int i=1;i<=n;i++)   add(i,1);
 for(int i=n;i;i--){
  int k=h[i]+1;
  int l=1,r=n;
  while(l<r){
   int mid=l+r>>1;
   if(sum(mid)>=k)    r=mid;
   else    l=mid+1;
  }
  ans[i]=r;
  add(r,-1);
 }
 for(int i=1;i<=n;i++)   printf("%d\n",ans[i]);
 return 0;
}