Cod sursa(job #2544791)

Utilizator StanCatalinStanCatalin StanCatalin Data 12 februarie 2020 15:28:01
Problema Heapuri Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream in("heapuri.in");
ofstream out("heapuri.out");

const int dim = 200005;

int n,v[dim],h[dim],poz[dim],nh,cnt,nr;

void Schimb(int p,int q)
{
    swap(h[p], h[q]);
    poz[h[p]] = p;
    poz[h[q]] = q;
}

void Urca(int p)
{
    while (p > 1 && v[h[p/2]] > v[h[p]])
    {
        Schimb(p, p/2);
        p /= 2;
    }
}

void Adauga(int p)
{
    h[++nh] = p;
    poz[p] = nh;
    Urca(nh);
}

void Coboara(int p)
{
    int fs = 2*p;
    int fd = 2*p + 1;
    int bun = p;
    if (fs <= nh && h[v[fs]] < h[v[bun]])
    {
        bun = fs;
    }
    if (fd <= nh && h[v[fd]] < h[v[bun]])
    {
        bun = fd;
    }
    if (bun != p)
    {
        Schimb(p, bun);
        Coboara(bun);
    }
}

void Sterge(int p)
{
    p = poz[p];
    Schimb(p , nh--);
    Urca(p);
    Coboara(p);
}

int main()
{
    in >> n;
    for (int i=1,c; i<=n; i++)
    {
        in >> c;
        if (c == 1)
        {
            in >> nr;
            v[++cnt] = nr;
            Adauga(cnt);
        }
        if (c == 2)
        {
            in >> nr;
            Sterge(nr);
        }
        if (c == 3)
        {
            out << v[h[1]] << "\n";
        }
    }
    return 0;
}