谜一样的牛（树状数组 + 维护前缀和）

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

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

输入格式
第 1 行：输入整数 n。

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

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

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

数据范围
1≤n≤10^5
在本题中，我们首先考虑最后一头牛。由于它已经统计了在它之前的所有牛，因此，假如它比x头牛高，则它的高度一定是x+1。我们采取从后往前考虑的方式，就是因为题目给出了“每头牛已知在自己前面比自己矮的牛的个数”这一条件，从后往前可以少考虑很多问题。

由于每头牛的高度各不相同且取遍区间[1,n]，因此，对于倒数第二头牛而言，它应该在除去上述x+1的区间[1,n]中，选取$A_{n-1}+1$小的数。其他的牛以此类推。

假如建立一个全部元素为1的数列，某个位置的数为1代表这个高度还不知道是哪头牛的，那么就用树状数组维护该数列的前缀和，若某个位置的前缀和等于$A_i+1$，此时的下标就是要找的数。选择这个数后，将相应位置的1置0.可以二分这个位置。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll inf=2e10;
const int maxn = 2e6+5;
const ll mp=5e9;
int n,c[maxn],a[maxn],ans[maxn];

int lowbit(int x){
	return x&(-x);
}
int sum(int x){
	int res=0;
	while(x){
		res+=c[x];
		x-=lowbit(x);
	}
	return res;
}
void updata(int x,int y){
	while(x<=n){
		c[x]+=y;
		x+=lowbit(x);
	}
}
int main(){
	scanf("%d",&n);
	for(int i=2;i<=n;i++){
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n;i++)c[i]=lowbit(i);
	for(int i=n;i;i--){
		int l=1,r=n,d=0;
		while(l<=r){
			int mid=(l+r)>>1;
			if(sum(mid)==a[i]+1){
				ans[i]=mid;
				r=mid-1;
		///		break;
			}
			else if(sum(mid)<a[i]+1)l=mid+1;
			else r=mid-1;
		}
		
		updata(ans[i],-1);
	}
	for(int i=1;i<=n;i++)
	    printf("%d\n",ans[i]);
}