poj 2182 树状数组

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

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

const int N = 8001;
int c[N];
int r[N];
int ans[N];
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
", ans[i]);
        }
    }
    return 0;
}