Cod sursa(job #213834)

Utilizator Mishu91Andrei Misarca Mishu91 Data 11 octombrie 2008 19:58:16
Problema Schi Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.12 kb
//varianta cu arbori indexati binar
#include <cstdio>

#define step ((k ^(k-1)) & k)
#define MAX_N 30005

int V[MAX_N], A[MAX_N], Sol[MAX_N];
int N;

void citire()
{
    scanf("%d",&N);

    for(int i = 1; i <= N; ++i)
        scanf("%d",V+i);
}

void update(int k, int val)
{
    while(k <= N)
    {
        A[k] += val;
        k += step;
    }
}

int querry(int sum)
{
    int i, pas;
    int x = sum, pmin = MAX_N;
    for(pas = 1; pas < N; pas <<= 1);

    for(i = 0; pas; pas >>= 1)
        if(i + pas <= N)
            if(sum >= A[i + pas])
            {
                i += pas, sum -= A[i];
                if((!sum) && i < pmin)
                    pmin = i, sum = x, i -= pas;
            }
    return pmin;
}

void solve()
{
    for(int i = 1; i <= N; ++i)
        update(i, 1);

    for(int i = N; i; i--)
    {
        int q = querry(V[i]);
        Sol[q] = i;
        update(q,-1);
    }
    for(int i = 1; i <= N; ++i)
        printf("%d\n",Sol[i]);
}

int main()
{
    freopen("schi.in","rt",stdin);
    freopen("schi.out","wt",stdout);
    citire();
    solve();
}