Cod sursa(job #2908419)

Utilizator _andrei4567Stan Andrei _andrei4567 Data 3 iunie 2022 12:16:08
Problema Schi Scor 85
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.15 kb
///mereu va fi al catalea zice din pozitiile ramase(incepem de la final)
#include <fstream>
#define lsb(x) x & (-x)

using namespace std;

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

const int N = 3e4;
int aib[N + 2], v[N + 2], sol[N + 2];

int n;
void update (int pos, int val)
{
    for (int i = pos; i <= n; i += lsb(i))
        aib[i] += val;
}

int query(int pos)
{
    int sum = 0;
    for (int i = pos; i >= 1; i -= lsb(i))
        sum += aib[i];
    return sum;
}

int cb (int st, int dr, int val)
{
    int med, last = -1;
    while (st <= dr)
    {
        med = (st + dr) >> 1;
        if (med - query(med) >= val)
        {
            last = med;
            dr = med - 1;
        }
        else
            st = med + 1;
    }
    return last;
}

int main()
{
    cin >> n;
    for (int i = 1; i <= n; ++i)
    {
        cin >> v[i];
    }
    sol[v[n]] = n;
    update (v[n], 1);
    for (int i = n - 1; i >= 1; --i)
    {
        int pos = cb(1, n, v[i]);
        update (pos, 1);
        sol[pos] = i;
    }
    for (int i = 1; i <= n; ++i)
        cout << sol[i] << '\n';

    return 0;
}