Cod sursa(job #2017663)

Utilizator savigunFeleaga Dragos-George savigun Data 1 septembrie 2017 23:34:31
Problema Schi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.19 kb
#include <fstream>
using namespace std;

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

typedef unsigned short int shi;

shi n, v[30001];
shi tree[120000];


void Update(shi pos, shi val, shi node = 1, shi l = 1, shi r = n) {
    if (l == r) {
        tree[node] = val;
        return;
    }

    shi mid = (l + r) / 2;

    if (pos <= mid) {
        Update(pos, val, node * 2, l, mid);
    } else {
        Update(pos, val, node * 2 + 1, mid + 1, r);
    }

    tree[node] = tree[node * 2] + tree[node * 2 + 1];
}

shi Query(shi kth, shi node = 1, shi l = 1, shi r = n) {
    if (l == r) return l;

    shi mid = (l + r) / 2;

    if (tree[node * 2] >= kth) {
        return Query(kth, node * 2, l, mid);
    } else {
        return Query(kth - tree[node * 2], node * 2 + 1, mid + 1, r);
    }
}



int main()
{
    shi order[30001];

    in >> n;

    for (int i = 1; i <= n; ++i) {
        in >> v[i];
        Update(i, 1);
    }

    for (int i = n; i >= 1; --i) {
        int pos = Query(v[i]);
        order[pos] = i;
        Update(pos, 0);
    }

    for (int i = 1; i <= n; ++i) {
        out << order[i] << "\n";
    }

    return 0;
}