Cod sursa(job #3250521)

Utilizator PsyDuck1914Feraru Rares-Serban PsyDuck1914 Data 21 octombrie 2024 16:34:42
Problema Schi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.67 kb
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

ifstream fin("schi.in");
ofstream fout("schi.out");

const int NMAX=30001;
int arb[NMAX*4];

struct OUTPUT{
    int index,position;
}K[NMAX];

bool cmp(OUTPUT x,OUTPUT y){
    return (x.position<y.position);
}

void generate_tree(int node,int st,int dr){
    if(st==dr){
        arb[node]=1;
    }
    else{
        int mid=(st+dr)/2;
        generate_tree(node*2,st,mid);
        generate_tree(node*2+1,mid+1,dr);
        arb[node]=arb[node*2]+arb[node*2+1];
    }
}

void update(int node,int st,int dr,int poz){
    if(st==dr && st==poz){
        arb[node]=0;
    }
    else{
        int mid=(st+dr)/2;
        if(poz<=mid)
            update(node*2,st,mid,poz);
        else
            update(node*2+1,mid+1,dr,poz);
        arb[node]=arb[node*2]+arb[node*2+1];
    }
}

int query(int node,int st,int dr,int sum){
    if(sum==1 && st==dr){
        return st;
    }
    else{
        if(sum<=arb[node*2]){
            int mid=(st+dr)/2;
            return query(node*2,st,mid,sum);
        }
        else{
            sum-=arb[node*2];
            int mid=(st+dr)/2;
            return query(node*2+1,mid+1,dr,sum);
        }
    }
}

int main()
{
    int n,poz;
    int v[NMAX];
    fin>>n;
    for(int i=1;i<=n;++i){
        fin>>v[i];
    }
    reverse(v+1,v+n+1);
    generate_tree(1,1,n);
    for(int i=1;i<=n;++i){
        poz=query(1,1,n,v[i]);
        update(1,1,n,poz);
        K[i].index=n-i+1;
        K[i].position=poz;
    }
    sort(K+1,K+n+1,cmp);
    for(int i=1;i<=n;++i){
        fout<<K[i].index<<"\n";
    }
    return 0;
}