Cod sursa(job #229881)

Utilizator MariusMarius Stroe Marius Data 11 decembrie 2008 23:43:28
Problema Schi Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.45 kb
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;

const char iname[] = "schi.in";
const char oname[] = "schi.out";

#define getc(n)  (n->cnt = (!n->el) + n->l->cnt + n->r->cnt)

struct T {
    int key, cnt, el;
    T *l, *r;
    T() {}
    T(int key, T *l, T *r) {
        this->key = key, this->l = l, this->r = r;
        cnt = el = 0;
    }
} *R, *nil;


void init(T* &R)
{
    srand(unsigned(time(NULL)));
    R = nil = new T(0, NULL, NULL);
}

void insert(T* &n, int lo, int hi)
{
    if (hi < lo)  return ;
    int mid = (hi + lo) >> 1;
    n = new T(mid, nil, nil);
    insert(n->l, lo, mid - 1);
    insert(n->r, mid + 1, hi);
    getc(n);
}

void lookup(T* &n, int cnt, int el)
{
    if (cnt <= n->l->cnt)
        lookup(n->l, cnt, el);
    else {
        if (n->l->cnt + (!n->el) == cnt)
            n->el = el;
        else
            lookup(n->r, cnt - (n->l->cnt + (!n->el)), el);
    }
    getc(n);
}

void print(T* &n) {
    if (n == nil)  return ;
    print(n->l), printf("%d\n", n->el), print(n->r);
}

int main(void)
{
    init(R);
    freopen(iname, "r", stdin);
    vector <int> V;
    int n;

    scanf("%d", &n);
    V.resize(n);
    for (int i = 0; i < n; ++ i)
        scanf("%d", &V[i]);
    insert(R, 0, n - 1);
    for (int i = n - 1; i >= 0; -- i)
        lookup(R, V[i], i + 1);

    freopen(oname, "w", stdout);
    print(R);

    return 0;
}