Lost Cows（线段树）

Lost Cows

题意：给出n头牛前面有多少头比他编号少的数目，求出原来的牛的编号，

思路：从后面开始遍历，碰到的a表示此牛在剩余的牛中排名a+1.线段树维护区间里面未标记的牛的数量，找到满足条件的区间即可。找到后同时更新一下。

#include<cstdio>
int sum[32005];
int a[8005],ql,qr,b[8000];
void build(int o,int l,int r){
	if(l==r){
		sum[o]=1;
		return;
	}
	int m=(l+r)>>1;
	build(o*2,l,m);
	build(o*2+1,m+1,r);
	sum[o]=sum[o*2]+sum[o*2+1];
}
int qu(int o,int l,int r){
	int m=(l+r)>>1;
	if(l==r)
		return r;
	else if(sum[o*2]>=ql){
		return qu(o*2,l,m);
	}
	else{
		ql-=sum[o*2];
		return qu(o*2+1,m+1,r);
	}
}
void up(int o,int l,int r){
	if(l==r){
		sum[o]--;
		return;
	}
	int m=(l+r)>>1;
	if(ql>m)
		up(o*2+1,m+1,r);
	if(ql<=m)
		up(o*2,l,m);
	sum[o]=sum[o*2]+sum[o*2+1];
}
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<n;i++)
		scanf("%d",&a[i]);
	a[0]=0;
	build(1,1,n);
	for(int i=n-1;i>=0;i--){
		ql=a[i]+1;
		b[i]=qu(1,1,n);
		ql=b[i];
		up(1,1,n);
	}
	for(int i=0;i<n;i++)
		printf("%d\n",b[i]);
}

暴力解法

include<cstdio>
int a[8005],vis[8005],b[8005];
int main(){
	int n;
	scanf("%d",&n);
	a[1]=0;
	for(int i=2;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=n;i>=1;i--){
		int t=a[i]+1;
		for(int j=1;j<=t;j++)
			if(vis[j])
				t++;
		b[i]=t;
		vis[t]=1;
	}
	for(int i=1;i<=n;i++)
		printf("%d\n",b[i]);
}