Cod sursa(job #2561832)

Utilizator denmirceaBrasoveanu Mircea denmircea Data 29 februarie 2020 10:33:54
Problema Schi Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <fstream>
#define dim 30005
using namespace std;
ifstream fin("schi.in");
ofstream fout("schi.out");
int n,v[dim],i,poz;
short aib[dim],sol[dim];
inline int query(int poz)
{
    int r=0;
    for(; poz; poz-=poz&-poz)
        r+=aib[poz];
    return r;
}
inline void update(int poz)
{
    for(; poz<=n; poz+=poz&-poz)
        aib[poz]++;
}
inline int solve(int x)
{
    int st=1-x,dr=n-x,mid;
    while(st<=dr)
    {
        mid=(st+dr)/2;
        if(mid>=query(mid))
            dr=mid-1;
        else st=mid+1;
    }
    return st+x;
}
int main()
{
    fin>>n;
    for(i=1; i<=n; i++)
        fin>>v[i];
    for(i=n; i>=1; i--)
    {
        poz=solve(v[i]);
        sol[poz]=i;
        update(poz);
    }
    for(i=1; i<=n; i++)
        fout<<sol[i]<<"\n";
}