Lost Cows[树状数组+二分]

传送门

我们倒过来做,对于当前的x,他前面有dx个比它小,那么它就是当前的第k大

这个第k大我们二分一下就好了,这个数完了过后,把它在树状数组里的值改为0就可以了

---

#include<cstdio>
#define N 8050
using namespace std;
int c[N],n,d[N],a[N];
void Up(int x,int val){for(;x<=n;x+=x&-x)c[x]+=val;}
int Q(int x){int ans=0;for(;x;x-=x&-x)ans+=c[x];return ans;}
int K(int x){
	int l=1,r=n;
	while(l<r){
		int mid=(l+r)>>1;
		if(Q(mid)>=x) r=mid;
		else l=mid+1;
	}return l;
}
int main(){
	scanf("%d",&n);
	for(int i=2;i<=n;i++) scanf("%d",&d[i]);
	for(int i=1;i<=n;i++) Up(i,1);
	for(int i=n;i>=1;i--){
		a[i]=K(d[i]+1); Up(a[i],-1);
	} 
	for(int i=1;i<=n;i++) printf("%d\n",a[i]); return 0;
}