AcWing244. 谜一样的牛（树状数组）

输入样例：

5
1
2
1
0

输出样例：

2
4
5
3
1

 解析：

        从后往前遍历，每次需要在剩余的数中，找到第 h[ i ]+1 大的数即为当前牛的身高。

        每次二分，然后求前缀和。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
ll n,m,h[N],ans[N],tr[N];
int lowbit(int x){
	return x&-x;
}
void add(int x,int k){
	for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=k;
}
int sum(int x){
	int ans=0;
	for(int i=x;i;i-=lowbit(i)) ans+=tr[i];
	return ans;
}
int main(){
	scanf("%d",&n);
	for(int i=2;i<=n;i++) scanf("%lld",&h[i]);
	for(int i=1;i<=n;i++) add(i,1);	
	for(int i=n;i>0;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(l,-1);
	}
	for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
	return 0;
}