Cod sursa(job #1400640)

Utilizator danalex97Dan H Alexandru danalex97 Data 25 martie 2015 13:05:40
Problema Schi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <cstdio>
#include <cstdlib>
using namespace std;

const int N = 30010;

struct item {
    int pr,vl,sz;
    item *l,*r;

    item(int vl)
    {
        this->vl = vl;
        pr = (rand() << 6) ^ rand();
        sz = 1;
        l = r = NULL;
    }
};

#define tree item*

void upd(tree &t)
{
    if ( !t ) return;
    t->sz = ( t->l ? t->l->sz : 0 ) + ( t->r ? t->r->sz : 0 ) + 1;
}

void split(tree t,int pl,tree &l,tree &r)
{
    if ( !t )
        return void(l = r = NULL);
    int sz = t->l ? t->l->sz : 0;
    if ( pl <= sz )
        split(t->l,pl,l,t->l) , r = t;
    else
        split(t->r,pl-sz-1,t->r,r) , l = t;
    upd(l);
    upd(r);
}

void insert(tree &t,int pl,tree x)
{
    if ( !t )
        return void(t = x);
    int sz = t->l ? t->l->sz : 0;
    if ( t->pr < x->pr )
        split(t,pl,x->l,x->r) , t = x;
    else
        if ( pl <= sz )
            insert(t->l,pl,x);
        else
            insert(t->r,pl-sz-1,x);
    upd(t);
}

void print(tree t)
{
    if ( t->l ) print(t->l);
    printf("%d\n",t->vl);
    if ( t->r ) print(t->r);
}

tree t = NULL;

int n,x;

int main()
{
    freopen("schi.in","r",stdin);
    freopen("schi.out","w",stdout);

    scanf("%d",&n);
    for (int i=1,p;i<=n;++i)
    {
        scanf("%d",&p);
        tree x = new item(i);
        insert(t,p-1,x);
    }
    print(t);
}