Cod sursa(job #1327316)

Utilizator MacWonkMihai Alexandru Cosmin MacWonk Data 26 ianuarie 2015 16:41:35
Problema Schi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <cstdio>
#define lb(x) (x&(-x))
#define nmax 30010
using namespace std;
int aib[nmax];
int pos[nmax], rez[nmax];
int i, n, x;
void minus(int x)
{
    for(int i=x; i<=n; i+=lb(i)) --aib[i];
}
int sum(int pos)
{
    int s=0;
    for(int i=pos; i>0; i-=lb(i)) s+=aib[i];
    return s;
}
int bs(int x)
{
    int l=1, r=n, m, cati, Min=999999;

    while(l<=r)
    {
        m=(l+r)/2;
        cati=sum(m);
        if(cati==x && m<Min) Min=m;
        else if(cati>=x) r=m-1;
            else l=m+1;
    }
    return Min;
}
int main()
{
    freopen("schi.in", "rt", stdin);
    freopen("schi.out", "wt", stdout);
    scanf("%d", &n);
    for(i=1;i<=n;++i) scanf("%d", &pos[i]);
    for(i=1; i<=n; ++i) aib[i]=lb(i);
    for(i=n; i>=1; --i)
    {
        x=bs(pos[i]);
        rez[x]=i;
        minus(x);
    }
    for(i=1; i<=n; ++i) printf("%d\n", rez[i]);
    return 0;
}