Cod sursa(job #3151352)

Utilizator SerbanCaroleSerban Carole SerbanCarole Data 20 septembrie 2023 19:47:33
Problema Schi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <fstream>
#include <algorithm>
#include <vector>
using namespace std;
ifstream cin("schi.in");
ofstream cout("schi.out");
int segt[4*30001] , n , ans[30001] , x;
vector <int> poz;
void build( int node , int st , int dr)
{
    segt[node] = dr - st + 1;
    if(st==dr) return;
    else
    {
        int mij = (st+dr)/2;
        build(node*2,st,mij);
        build(node*2+1,mij+1,dr);
    }
}
void upd( int node , int st , int dr , int &poz)
{
    if(st==dr) segt[node] = 0;
    else
    {
        int mij = (st+dr)/2;
        if(poz<=mij) upd(node*2,st,mij,poz);
        else upd(node*2+1,mij+1,dr,poz);
        segt[node]=segt[node*2]+segt[node*2+1];
    }
}
int bs( int node , int st , int dr , int poz)
{
    if(st==dr) return st;
    else
    {
        int mij = (st+dr)/2;
        if(segt[node*2] >= poz) return bs(node*2,st,mij,poz);
        else
        {
            poz -= segt[node*2];
            return bs(node*2+1,mij+1,dr,poz);
        }
    }
}
signed main()
{
    cin >> n;
    build(1,1,n);
    for(int i = 1 ; i <= n ; i++)
    {
        cin >> x; poz.push_back(x);
    }
    reverse(poz.begin(),poz.end());
    int wo = n;
    for(auto &it : poz)
    {
        int val = bs(1,1,n,it);
        ans[val] = wo;
        wo--;
        upd(1,1,n,val);
    }
    for(int i = 1 ; i <= n ; i++) cout << ans[i] << '\n';
    return 0;
}