Cod sursa(job #2536591)

Utilizator cyg_vladioanBirsan Vlad cyg_vladioan Data 2 februarie 2020 12:20:13
Problema Schi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.15 kb
#include <cstdio>
using namespace std;
const int NMAX = 30000;
int v[NMAX + 5] , aib[NMAX + 5] , ans[NMAX + 5] , n;
void update(int poz)
{
    for( ; poz <= n ; poz = poz + (poz & (-poz)))
        aib[poz] ++;
}
int query(int poz)
{
    int suma = 0;
    for( ; poz > 0 ; poz = poz - (poz & (-poz)))
        suma = suma + aib[poz];
    return suma;
}
int bs(int k)
{
    int st , dr , med , last , aux;
    st = k;
    dr = n;
    last = -1;
    while(st <= dr)
    {
        med = (st + dr) / 2;
        aux = query(med);
        if(aux + k == med)
        {
            last = med;
            dr = med - 1;
        }
        else if(aux + k < med)
            dr = med - 1;
        else
            st = med + 1;
    }
    return last;
}
int main()
{
    freopen("schi.in" , "r" , stdin);
    freopen("schi.out" , "w" , stdout);
    int i , poz;
    scanf("%d" , &n);
    for(i = 1 ; i <= n ; i ++)
        scanf("%d" , &v[i]);
    for(i = n ; i >= 1 ; i --)
    {
        poz = bs(v[i]);
        update(poz);
        ans[poz] = i;
    }
    for(i = 1 ; i <= n ; i ++)
        printf("%d\n" , ans[i]);
    return 0;
}