Cod sursa(job #2903854)

Utilizator NicuDirvaDirva Nicolae NicuDirva Data 17 mai 2022 21:01:31
Problema Schi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.33 kb
#include <iostream>
#include <fstream>
using namespace std;

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

int v[30001], d[120001], clvs[30001];

void crevre(int nod, int st, int dr)
{
    if(st == dr)
    {
        d[nod] = v[st];
        return;
    }
    crevre(nod * 2, st, (st + dr) / 2);
    crevre(nod * 2 + 1, (st + dr) / 2 + 1, dr);
    d[nod] = d[nod * 2] + d[nod * 2 + 1];
}

int fixvre(int nod, int st, int dr, int poz_interm)
{
    if(st == dr)
        return st;
    if(d[2*nod] < poz_interm)
        return fixvre(nod * 2 + 1, (st + dr) / 2 + 1, dr, poz_interm - d[nod * 2]);
    return fixvre(nod * 2, st, (st + dr) / 2, poz_interm);
}

void modificare(int nod, int st, int dr, int i, int vvl)
{
    if(i < st || i > dr) return;
    if(st == dr)
    {
        d[nod] = vvl;
        return;
    }
    modificare(nod * 2, st, (st + dr) / 2, i, vvl);
    modificare(nod * 2 + 1, (st + dr) / 2 + 1, dr, i, vvl);
    d[nod] = d[nod * 2] + d[nod * 2 + 1];
}

int main()
{int n, p;

fin >> n;

for(int i = 1; i <= n; i++)
    v[i] = 1;

crevre(1, 1, n);

for(int i = 1; i <= n; i++)
    fin >> v[i];

for(int i = n; i >= 1; i--)
{
    p = fixvre(1, 1, n, v[i]);
    clvs[p]=i;
    modificare(1, 1, n, p, 0);
}

for(int i = 1; i <= n; i++)
    fout << clvs[i] << "\n";

return 0;
}