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