poj 2182 树状数组

倒着考虑，如果最后一只牛的前面有x只比它小，那么它就是第x+1只牛，从序列中去掉它。对倒数第二只牛来说也同理。可以用树状数组来维护前缀和，一开始每个位置都是1，求出结果的牛从树状数组中删掉（update成0），即可获得答案。

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

const int N = 8001;
int c
;
int r
;
int ans
;
int n;

int lb( int i )
{
return i & -i;
}

void update( int i, int v )
{
while ( i <= n )
{
c[i] += v;
i += lb(i);
}
}

int kth( int k )
{
int ans = 0, cnt = 0;
for ( int i = 12; i >= 0; i-- )
{
ans += ( 1 << i );
if ( ans >= n || cnt + c[ans] >= k )
{
ans -= ( 1 << i );
}
else
{
cnt += c[ans];
}
}
return ans + 1;
}

int main ()
{
while ( scanf("%d", &n) != EOF )
{
memset( c, 0, sizeof(c) );
for ( int i = 1; i <= n; i++ )
{
update( i, 1 );
}
r[1] = 1;
for ( int i = 2; i <= n; i++ )
{
scanf("%d", &r[i]);
r[i]++;
}
for ( int i = n; i >= 1; i-- )
{
ans[i] = kth( r[i] );
update( ans[i], -1 );
}
for ( int i = 1; i <= n; i++ )
{
printf("%d\n", ans[i]);
}
}
return 0;
}