1. 程式人生 > >Lost Cows[樹狀陣列+二分]

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;
}