Cod sursa(job #2832372)

Utilizator DordeDorde Matei Dorde Data 13 ianuarie 2022 15:38:38
Problema Schi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.94 kb
#include<bits/stdc++.h>
using namespace std;
ifstream fin("schi.in");
ofstream fout("schi.out");
int const N = 3e4 + 3;
int n , p[N] ;
map<int , int> ans;
struct aib{
    #define z(x) ((x)&(-x))
    int v[N];
    aib(){
        fill(v , v + N , 0);
    }
    void add(int p , int x){
        for(int i = p ; i <= n ; i += z(i))
            v[i] += x;
    }
    int query(int x){
        int pas = (1 << 14) , r = 0;
        while(pas){
            if(pas + r <= n && v[pas + r] < x){
                x -= v[pas + r];
                r += pas;
            }
            pas >>= 1;
        }
        return r + 1;
    }
}t;
int main()
{
    fin >> n;
    for(int i = 1 ; i <= n ; ++ i){
        fin >> p[i];
        t.add(i , 1);
    }
    for(int i = n ; i ; -- i){
        int r = t.query(p[i]);
        t.add(r , -1);
        ans[r] = i;
    }
    for(auto [a , b] : ans)
        fout << b << '\n';
    return 0;
}