Cod sursa(job #1520926)

Utilizator LipanmateiLipan Radu-Matei Lipanmatei Data 9 noiembrie 2015 18:39:13
Problema Schi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.06 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("schi.in");
ofstream fout("schi.out");
int ARB[90003], Rasp[30001], v[30001],n;
void add(int nod, int st, int dr,int val,int poz)
{
    if(st==dr)
    {
        ARB[nod]+=val;
        return;
    }
    else
    {
        int mij=(st+dr)>>1;
        if(poz<=mij)add(nod<<1,st,mij,val,poz);
        else add((nod<<1)+1,mij+1,dr,val,poz);
    }
    ARB[nod]=ARB[nod<<1]+ARB[(nod<<1)+1];
}
int srch(int val)
{
    int nod=1, st=1, dr=n;
    while(st<dr)
    {
        if(ARB[nod<<1]>=val)
        {
            dr=(st+dr)>>1;
            nod<<=1;
        }
        else
        {
            st=((st+dr)>>1)+1;
            val-=ARB[nod<<1];
            nod=(nod<<1)+1;
        }
    }
    return st;
}
int main()
{
    fin>>n;
    for(int i=1;i<=n;i++)
        fin>>v[i],add(1,1,n,1,i);
    for(int i=n;i>=1;i--)
    {
        int poz=srch(v[i]);
        Rasp[poz]=i;
        add(1,1,n,-1,poz);
    }
    for(int i=1;i<=n;i++)fout<<Rasp[i]<<'\n';
    return 0;
}