Cod sursa(job #1276008)

Utilizator sebinechitasebi nechita sebinechita Data 25 noiembrie 2014 21:19:21
Problema Schi Scor 85
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.73 kb
#include <fstream>
#include <cstdlib>
std::ifstream fin("schi.in");
std::ofstream fout("schi.out");

struct T{
    int key, priority;
    T *left, *right;

    void update()
    {
        int s=1;
        if(left)
            s+=(left->key&((1<<15)-1));
        if(right)
            s+=(right->key&((1<<15)-1));
        key=((key>>15)<<15)+s;
    }
    T(int k, int p, T* l, T* r)
    {
        priority = p;
        key = k<<15;
        left = l;
        right = r;
    }
} *R, *nil;

void init()
{
    srand(0);
    R = nil = new T(0, 0, 0, 0);
}

void rotL(T* &n)
{
    T* x = n->left;
    n->left = x->right;
    x->right = n;
    n->update();
    n = x;
    n->update();
}

void rotR(T* &n)
{
    T* x = n->right;
    n->right = x->left;
    x->left = n;
    n->update();
    n = x;
    n->update();
}

void balance(T* &n)
{
    if(n->left->priority > n->priority)
        rotL(n);
    else if(n->right->priority > n->priority)
        rotR(n);
}

void insert(T* &n, int key, int priority, float value)
{
    if(n == nil)
    {
        n = new T(key, priority, nil, nil);
        n->update();
        return;
    }
    if((n->left->key&((1<<15)-1)) + 1 > value)
    {
        insert(n->left, key, priority, value);
        n->update();
    }
    else
    {
        insert(n->right, key, priority, value-((n->left->key&((1<<15)-1))+1));
        n->update();
    }
    balance(n);
}

void af(T* &n)
{
    if(n == nil)
        return;
    af(n->left);
    fout << (n->key>>15) << "\n";
    af(n->right);
}

int main()
{
    int n, i, x;
    init();
    fin>>n;
    for(i=1; i<=n; i++)
    {
        fin>>x;
        insert(R, i, rand()+1, x-0.5);
    }
    af(R);
}