Cod sursa(job #2842374)

Utilizator AndreiCroitoruAndrei Croitoru AndreiCroitoru Data 31 ianuarie 2022 18:12:01
Problema Heapuri Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <bits/stdc++.h>

using namespace std;

int v[200001], h[200001], poz[200001];

int n, nh;

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]]<v[h[p/2]])
    {
        schimb (p,p/2);
        p=p/2;
    }
}

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

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

void sterge(int p)
{
    if (p==nh)
    {
        nh--;
    }
    else
    {
        schimb(nh--,p);
        urca(p);
        coboara(p);
    }
}

int main()
{

    int t;
    cin >> t;
    while(t--)
    {
        int q, nod;
        cin >> q;
        if (q == 1)
        {
            cin >> v[++n];
            adauga(n);
        }
        if (q == 2)
        {
            int x;
            cin >> x;
            sterge(poz[x]);
        }
        else
        {
            cout << v[h[1]] << '\n';
        }
    }
    return 0;
}