POJ 2182 Lost Cows 线段树

题意: 将N头进行编号，1 -- n。现在所有的牛随机站成一条队列， 我们仅仅知道第 i 个位置的牛前面，有 j 头牛的编号比它的小。要求每个位置的牛的编号。
输入的第一行是一个整数N, 接着有N-1行输入，每一行一个整数。假设第5行的数字为3，则表示第5个位置的的牛前面有3头牛的编号比它小。（因为第一个位置的牛的前面比它编号小的肯定没有，故第一个位置的输入被忽略了，所以输入N之后只输入了N-1行）。

题解：此题和 poj2828 Buy Tickets 有几分相似。

#include <iostream>
using namespace std;

#define L(u) ( u << 1 )
#define R(u) ( u << 1 | 1 )
#define N 8010

int a[N], num[N];

struct item
{
	int l, r, v; /* v表示了节点内部的容量 */
} node[N*3];

void build ( int u, int l, int r )
{
	node[u].l = l;
	node[u].r = r;
	node[u].v = r - l + 1;
	if ( l == r ) return;
	int mid = ( l + r ) >> 1;
	build ( L(u), l, mid );
	build ( R(u), mid + 1, r );
}

int query ( int u, int pos )
{
	node[u].v--;
	if ( node[u].l == node[u].r )
		return node[u].r;	
	if ( pos <= node[L(u)].v )
		return query ( L(u), pos );
	else
		return query ( R(u), pos - node[L(u)].v );
}

int main()
{
	int n, i;
	scanf("%d",&n);
	build ( 1, 1, n );
	for ( i = 1; i < n; i++ )
		scanf("%d",&a[i]);

	a[0] = 0;
	for ( i = n - 1; i >= 0; i-- )
		num[i] = query ( 1, a[i] + 1 );
	
	for ( i = 0; i < n; i++ )
		printf("%d\n",num[i]);
	return 0;
}