Cod sursa(job #849164)

Utilizator assa98Andrei Stanciu assa98 Data 6 ianuarie 2013 17:14:33
Problema Heapuri Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.94 kb
#include <cstdio>
#define fiu_s(x) (x)*2
#define fiu_d(x) (x)*2+1
#define tata(x) (x)/2

int k;
int h[200200];
int p[200200]; // nodul inserat al  i-lea se afla pe poz p[i] in heap
int sp[200200]; //pe poz i se afla nodul inserat al sp[i]-lea

int c;

int n;

void shvap(int nod,int fiu)
{
    int tmp;
    tmp=sp[nod];
    sp[nod]=sp[fiu];
    sp[fiu]=tmp;

    p[sp[nod]]=nod;
    p[sp[fiu]]=fiu;

    tmp=h[nod];
    h[nod]=h[fiu];
    h[fiu]=tmp;
}

void down(int nod)
{
    bool pp;
    do
    {
        pp=false;
        int fiu=nod;

        if(fiu_s(nod)<=n)
        {
            if(h[fiu]>h[fiu_s(nod)])
            {
                pp=true;
                fiu=fiu_s(nod);
            }
        }

        if(fiu_d(nod)<=n)
        {
            if(h[fiu]>h[fiu_d(nod)])
            {
                pp=true;
                fiu=fiu_d(nod);
            }
        }

        //schimb h[nod] cu h[fiu]
        if(pp)
        {
            shvap(nod,fiu);
            nod=fiu;
        }
    }
    while(pp);
}

void up(int nod)
{
    while(tata(nod)>=1&&h[nod]<h[tata(nod)])
    {
        shvap(nod,tata(nod));
        nod=tata(nod);
    }
}

void insert(int x)
{
    ++n;
    h[n]=x;
    p[c]=n;
    sp[n]=c;
    up(n);
}

void del(int x)
{
    int nod=p[x];
    shvap(nod,n);
    n--;
    if(tata(nod)>=1)
    {
        up(nod);
    }
    if(fiu_s(nod)<=n)
    {
        down(nod);
    }
}

int main()
{
    freopen("heapuri.in","r",stdin);
    freopen("heapuri.out","w",stdout);
    scanf("%d",&k);
    int t,x;
    for(int i=1;i<=k;i++)
    {
        scanf("%d",&t);
        if(t==3)
        {
            printf("%d\n",h[1]);
        }
        else if(t==1)
        {
            scanf("%d",&x);
            c++;
            insert(x);
        }
        else
        {
            scanf("%d",&x);
            del(x);
        }
    }
    return 0;
}