Cod sursa(job #2017667)

Utilizator savigunFeleaga Dragos-George savigun Data 1 septembrie 2017 23:42:19
Problema Schi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.2 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) >> 1;

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

    tree[node] = tree[node << 1] + tree[node << 1 | 1];
}

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

    shi mid = (l + r) >> 1;

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



int main()
{
    shi order[30001];

    in >> n;

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

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

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

    return 0;
}