Cod sursa(job #2560554)

Utilizator AlexPascu007Pascu Ionut Alexandru AlexPascu007 Data 28 februarie 2020 08:58:18
Problema Schi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.83 kb
#include <bits/stdc++.h>
#define DIM 30010
using namespace std;
ifstream fin("schi.in");
ofstream fout("schi.out");
int n,i,v[DIM],sol[DIM],A[DIM],poz;
void update(int poz,int val) {
    for (;poz<=n;poz+=(poz&-poz))
        A[poz]+=val;
}
int query(int poz) {
    int val=0;
    for (;poz;poz-=(poz&-poz))
        val+=A[poz];
    return val;
}
int cb(int val) {
    int st=1,dr=n,sol=n+1;
    while (st<=dr) {
        int mid=(st+dr)/2;
        if (val<=query(mid))
            dr=mid-1;
        else
            st=mid+1;
    }
    return st;
}
int main() {
    fin>>n;
    for (i=1;i<=n;i++) {
        fin>>v[i];
        update(i,1);
    }
    for (i=n;i>=1;i--) {
        poz=cb(v[i]);
        sol[poz]=i;
        update(poz,-1);
    }
    for (i=1;i<=n;i++)
        fout<<sol[i]<<"\n";
    return 0;
}