Cod sursa(job #3234774)

Utilizator rapidu36Victor Manz rapidu36 Data 11 iunie 2024 18:28:39
Problema Heapuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.6 kb
#include <fstream>

using namespace std;

const int N = 2e5;

int nv, nh, n, v[N], h[N], poz[N];

int tata(int p)
{
    return ((p - 1) / 2);
}

int fiu_s(int p)
{
    return (2 * p + 1);
}

int fiu_d(int p)
{
    return (2 * p + 2);
}

void schimba(int p1, int p2)
{
    swap(h[p1], h[p2]);
    poz[h[p1]] = p1;
    poz[h[p2]] = p2;
}

void urca(int p)
{
    while (p > 0 && v[h[p]] < v[h[tata(p)]])
    {
        schimba(p, tata(p));
        p = tata(p);
    }
}

void adauga(int pv)
{
    h[nh++] = pv;
    poz[pv] = nh - 1;
    urca(nh - 1);
}

void coboara(int p)
{
    int fs = fiu_s(p), fd = fiu_d(p), optim = p;
    if (fs < nh && v[h[fs]] < v[h[optim]])
    {
        optim = fs;
    }
    if (fd < nh && v[h[fd]] < v[h[optim]])
    {
        optim = fd;
    }
    if (optim != p)
    {
        schimba(p, optim);
        coboara(optim);
    }
}

void sterge(int ph)
{
    if (ph == nh - 1)
    {
        nh--;
        return;
    }
    schimba(ph, nh - 1);
    nh--;
    urca(ph);
    coboara(ph);
}

int main()
{
    ifstream in("heapuri.in");
    ofstream out("heapuri.out");
    in >> n;
    for (int i = 0; i < n; i++)
    {
        int tip;
        in >> tip;
        if (tip == 1)
        {
            in >> v[nv++];
            adauga(nv - 1);
        }
        else if (tip == 2)
        {
            int pv;
            in >> pv;
            pv--;
            sterge(poz[pv]);
        }
        else
        {
            out << v[h[0]] << "\n";
        }
    }
    in.close();
    out.close();
    return 0;
}