Cod sursa(job #1291171)

Utilizator valentin.harsanValentin Harsan valentin.harsan Data 12 decembrie 2014 14:05:03
Problema Schi Scor 75
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.72 kb
#include<iostream>
#include<fstream>
#include<time.h>
#include<cstdlib>
using namespace std;

const int N = 30100;

struct t {
    int k, p, nrel;
    t *f1, *f2;

    t() {
        k = p = 0;
        nrel = 0;
    }

    t(int a, int b, t *c, t *d) {
        k = a; p = b;
        f1 = c; f2 = d;
        nrel = 1;
    }
};

t *rad, *nil;

int n;
int x[N];

void rotLeft(t* &r) {
    t *temp = r->f1;
    r->f1 = temp->f2;
    temp->f2 = r;
    r = temp;
}

void rotRight(t* &r) {
    t *temp = r->f2;
    r->f2 = temp->f1;
    temp->f1 = r;
    r = temp;
}

void recalc(t* &r) {
    if(r != nil)
        r->nrel = 1 + r->f1->nrel + r->f2->nrel;
}

void balance(t* &r) {
    if(r->f2->p > r->p)
        rotRight(r);
    if(r->f1->p > r->p)
        rotLeft(r);

    recalc(r->f1);
    recalc(r->f2);
    recalc(r);
}

void update(t* &r, int key, int val, int pri) {
    if(r == nil) {
        r = new t(key, pri, nil, nil);
        return;
    }

    if(r->f1->nrel < val)
        update(r->f2, key, val - (r->f1->nrel) - 1, pri);
    else
        update(r->f1, key, val, pri);

    balance(r);
}

void print(t* &r) {
    if(r == nil)
        return;
    print(r->f1);
    cout << r->k << "\n";
    print(r->f2);
}

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

    srand(777);
    nil = new t;
    rad = new t;
    nil->f1 = nil->f2 = nil;

    cin >> n;
    cin >> x[1];
    rad->k = 1;
    rad->p = rand() + 1;
    rad->f1 = rad->f2 = nil;
    rad->nrel = 1;

    for(i = 2; i <= n; ++i) {
        cin >> x[i];
        update(rad, i, x[i] - 1, rand() + 1);
    }

    print(rad);

    return 0;
}