Cod sursa(job #3176160)

Utilizator Emilia23Dobra Emilia Emilia23 Data 26 noiembrie 2023 19:27:09
Problema Schi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.26 kb
#include <bits/stdc++.h>

using namespace std;

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

int n,v[30005],aint[120005],sol[30005];

void init(int nod,int st,int dr)
{
    if(st==dr)aint[nod]=1;
    else
    {
        int mij=(st+dr)/2;
        init(2*nod,st,mij);
        init(2*nod+1,mij+1,dr);
        aint[nod]=aint[2*nod]+aint[2*nod+1];
    }
}

void init()
{
    init(1,1,n);
}

void update(int nod,int st,int dr,int index)
{
    if(st==dr)aint[nod]=0;
    else
    {
        int mij=(st+dr)/2;
        if(index<=mij)update(2*nod,st,mij,index);
            else update(2*nod+1,mij+1,dr,index);
        aint[nod]=aint[2*nod]+aint[2*nod+1];
    }
}

void update(int index)
{
    update(1,1,n,index);
}

int query(int nod,int st,int dr,int k)
{
    if(st==dr)return st;
    else
    {
        int mij=(st+dr)/2;
        if(k<=aint[2*nod])return query(2*nod,st,mij,k);
        else return query(2*nod+1,mij+1,dr,k-aint[2*nod]);
    }
}

int query(int k)
{
    return query(1,1,n,k);
}

int main()
{
    int i,x;
    f>>n;
    for(i=1;i<=n;i++)f>>v[i];
    init();
    for(i=n;i>=1;i--)
    {
        x=query(v[i]);
        sol[x]=i;
        update(x);
    }
    for(i=1;i<=n;i++)g<<sol[i]<<'\n';
    return 0;
}