Cod sursa(job #751293)

Utilizator deneoAdrian Craciun deneo Data 25 mai 2012 14:24:25
Problema Schi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <fstream>
using namespace std;

ifstream fin("schi.in");
ofstream fout("schi.out");

#define MAXN 30100

int a[MAXN], n, arb[MAXN], rez[MAXN];

void update (int poz, int val) {
    while (poz <= n) {
        arb[poz] += val;
        poz += poz & (-poz);
    }
}

int query (int x) {
    int i, step;
    for (step = 1; step < n; step <<= 1);

    for (i = 0; step; step >>= 1) {
        if(i + step <= n && arb[i + step] < x) {
            x -= arb[i + step];
            i += step;
        }
    }
    return i + 1;
}

int main() {
    int i, r;
    fin >> n;
    for (i = 1; i <= n; ++i) {
        fin >> a[i];
        update(i, 1);
    }
    for (i = n; i > 0; --i) {
        r = query(a[i]);
        update(r, -1);
        rez[r] = i;
    }

    for (i = 1; i <= n; ++i)
        fout << rez[i] << "\n";
    return 0;
}