POJ--Lost Cows （线段树）

题目：http://poj.org/problem?id=2182    http://acm.hdu.edu.cn/showproblem.php?pid=2711

 

题意：有N头牛，编号为1--N。 乱序排成一列，已知每头牛前面有多少头牛比它的编号小（从第二头牛开始）。

现在需要求这个序列中从前到后，每一头牛的编号。

 

思路：因为有N头牛，编号为1--N，最后一头牛如果前面有K头牛比它小，那么可以知道最后这头牛的编号为K+1.

当最后一头牛编号为K+1确定以后，再来看倒数第二头牛，如果这头牛前面有K₁头牛比它小，那么这头牛的编号

必然是剩下N-1头牛中第（K₁+1）大的。然后以此类推。。。。

最开始的线段树为：

最后一头牛前面有0个比它小，那么这是找第（0+1）大的数。为左下角的（1，1）即1.

倒数第二头牛前面有1个比它小，那么这头牛是剩下所有牛中第（1+1）大的，为（3，3）即3

倒数第三头牛前面有2个比它大，那么这头牛是剩下所有牛中第（2+1）大的，为（5，5）即5.

 

代码：

#include<cstdio>
 
#include<cstring>
 
#define lson l,m,rt<<1
 
#define rson m+1,r,rt<<1|1
 
using namespace std;
 
const int maxn=8010;
 
int n,arr[maxn],sum[maxn<<2];
 
int num[maxn];
 
void build(int l,int r,int rt)
 
{
 
	sum[rt]=r-l+1;
 
	if(l==r) return;
 
	int m=(l+r)>>1;
 
	build(lson);
 
	build(rson);
 
}
 
int query(int p,int l,int r,int rt)
 
{
 
	sum[rt]--;
 
	if(l==r) return l;
 
	int m=(l+r)>>1;
 
	if(p<=sum[rt<<1]) return query(p,lson);
 
	else			  return query(p-sum[rt<<1],rson);
 
}
 
int main()
 
{
 
	while (~scanf("%d",&n))
 
	{
 
		build(1,n,1);
 
		for(int i=2;i<=n;i++) scanf("%d",&arr[i]);
 
		arr[1]=0;
 
		for(int i=n;i>0;i--)
 
			num[i]=query(arr[i]+1,1,n,1);
 
		for(int i=1;i<=n;i++)
 
			printf("%d\n",num[i]);
 
	}
 
	return 0;
 
}