Cod sursa(job #1064151)

Utilizator Dddarius95Darius-Florentin Neatu Dddarius95 Data 21 decembrie 2013 16:47:40
Problema Heapuri Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.13 kb
#include <fstream>
#include <algorithm>
#define Nmax 200099
#define Father(i) i/2
#define LeftSon(i) 2*i
#define RightSon(i) 2*i+1
using namespace std;
ifstream f("heapuri.in");
ofstream g("heapuri.out");

int N,n,H[Nmax],v[Nmax],w[Nmax],nr;
//nr va reprezenta cate chei au fost introduse
//cand o cheie dispare e nu se modifica!
//v[i]= nodul care a intrat al i-lea in ordine cronologica
//w[nod]= pe ce pozitie cronologica a intrat nodul nod


inline void Swapp(int x,int y)
{
    swap(w[x],w[y]);
    v[w[x]]=x;
    v[w[y]]=y;
    swap(H[x],H[y]);
}

void DownHeap(int H[], int n,int nod)
{
    int Son=0;
    do
    {
        Son=0;
        // Alege un fiu mai mare ca tatal.
        if (LeftSon(nod)<=n)
        {
            Son=LeftSon(nod);
            if( RightSon(nod)<=n && H[RightSon(nod)]<H[LeftSon(nod)] ) Son=RightSon(nod);
            if (H[Son]>=H[nod]) Son=0;
        }
        if (Son)
        {
            Swapp(nod,Son);
            nod=Son;
        }
    } while(Son);
}

inline void UpHeap(int H[], int n,int nod)
{
    int key=H[nod];
    while (nod>1 && key<H[Father(nod)])
    {
        Swapp(nod,Father(nod));
        nod=Father(nod);
    }
    H[nod]=key;
}


inline void DeleteHeap(int H[], int &n,int x)
{
    int nod=v[x];
    Swapp(v[x],n);
    --n;
    UpHeap(H,n,nod);
    DownHeap(H,n,nod);
}

inline void InsertHeap(int H[], int &n,int key)
{
    H[++n]=key;
    nr++;
    v[nr]=n;
    w[n]=nr;
    UpHeap(H,n,n);
}

int main()
{
    f>>N;
    for(int i=1;i<=N;i++)
    {
        int tip;
        f>>tip;
        switch (tip)
        {
            case 1:
            {
                int key;
                f>>key;
                InsertHeap(H,n,key);
                break;
            }
            case 2:
            {
                int x;
                f>>x;
                DeleteHeap(H,n,x);
                break;
            }
            case 3:
            {
                g<<H[1]<<'\n';
                break;
            }
            default: break;
        }
    }
    f.close();g.close();
    return 0;
}