Cod sursa(job #1745458)

Utilizator alexmisto342Turdean Alexandru alexmisto342 Data 21 august 2016 22:12:08
Problema Schi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.6 kb
#include <cstdio>
#include <ctime>
#include <cstdlib>
#define inf 2000000000
using namespace std;
//ifstream fin("schi.in");
//ofstream fout("schi.out");
int i,n,poz;
struct nod{
    short  info;
    int pri;short nr;
    nod *st, *dr;
    nod(int info, int pri, nod *st, nod *dr)
    {
        this->dr = dr;
        this->st = st;
        this->info = info;
        this->pri = pri;
        nr = 0;
    }
};
nod *T,*nil;

void upd(nod *&n)
{
    n->nr = n->st->nr + n->dr->nr + 1;
}
void rotateleft(nod *&n)
{
    nod *x = n->st;
    n->st = x->dr;
    x->dr = n;
    n = x;
}

void rotateright(nod *&n)
{
    nod *x = n->dr;
    n->dr = x->st;
    x->st = n;
    n = x;
}

void balance(nod *&n)
{
    if(n->pri < n->st->pri)
        rotateleft(n);
    else
        if(n->pri < n->dr->pri)
            rotateright(n);
    if(n->st!=nil)
    upd(n->st);
    if(n->dr!=nil)
    upd(n->dr);
    upd(n);
}

void insert(nod *&n, int info, int prioritate)
{
    if(n == nil)
    {
        n = new nod(info,prioritate,nil,nil);
        n->nr= 1;
        return;
    }

    if(poz <= n->st->nr)
        insert(n->st,info,prioritate);
    else
        poz-=n->st->nr+1,insert(n->dr,info,prioritate);

    balance(n);
}

void del(nod *&n,int info)
{
    if(n == nil)
        return;
    if(info > n->info)
        del(n->dr, info);
    else
    if(info < n->info)
        del(n->st, info);
    else
    if(n->st == nil && n->dr == nil){
        delete n, n = nil;return;}
    else
    {
        if(n->st->pri > n->dr->pri){
            rotateleft(n);
            del(n->dr,info);}
        else
            rotateright(n),del(n->st,info);

    }
    upd(n);
}
void split(nod *&T, nod *&Ts, nod *&Td, int info)
{
    insert(T,info,inf+100);
    Ts = T->st;
    Td = T->dr;
    delete T, T=nil;
}

void join(nod *&T, nod *&Ts, nod *&Td)
{
    T = new nod(0,0,Ts,Td);
    del(T,T->info);
}
int find(nod *&n,int k)
{
    if(n->st->nr < k)
    {
        k -= n->st->nr;
        if(k == 1)
            return n->info;
        else
            return find(n->dr,k-1);
    }
    else
        return find(n->st,k);

}
void afisare(nod *&n)
{
    if(n == nil)
        return;
    afisare(n->st);
    printf("%d\n",n->info);
    afisare(n->dr);
}
int main()
{
    srand(time(0));

    T = nil = new nod(0,0,nil,nil);
   freopen("schi.in","r",stdin);
    freopen("schi.out","w",stdout);

    scanf("%d",&n);
   for(i = 1; i <= n; i++)
   {
       scanf("%d",&poz);poz--;
       insert(T,i,rand()+1);
   }
   afisare(T);
    return 0;
}