Cod sursa(job #1233608)

Utilizator Theodor1000Cristea Theodor Stefan Theodor1000 Data 25 septembrie 2014 19:31:44
Problema Heapuri Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.78 kb
#include <cstdio>
#include <algorithm>

#define f first
#define s second

using namespace std;

int n, vv[400010];
pair <int, int> v[400010];

void heapup (int k)
{
    if (k == 1) return;
    if (v[k / 2].f > v[k].f)
    {
        swap (v[k / 2], v[k]);
        swap (vv[v[k].s], vv[v[k / 2].s]);
        heapup (k / 2);
    }
}

void heapdown (int k, int n)
{
    if (v[k * 2].f <= v[k * 2 + 1].f && k * 2 <= n)
    {
        if (v[k].f >= v[k * 2].f)
        {
            swap (v[k], v[k * 2]);
            swap (vv[v[k].s], vv[v[k * 2].s]);
            heapdown (2 * k, n);
        }
    }

    else if (v[k * 2].f >= v[k * 2 + 1].f && k * 2 + 1 <= n)
    {
        if (v[k].f >= v[k * 2 + 1].f)
        {
            swap (v[k], v[k * 2 + 1]);
            swap (vv[v[k].s], vv[v[k * 2 + 1].s]);
            heapdown (2 * k + 1, n);
        }
    }

    else if (k * 2 <= n && k * 2 + 1 > n)
    {
        if (v[k].f >= v[k * 2].f)
        {
            swap (v[k], v[k * 2]);
            swap (vv[v[k].s], vv[v[k * 2].s]);
            heapdown (2 * k, n);
        }
    }
}

int main ()
{
    freopen ("heapuri.in", "r", stdin);
    freopen ("heapuri.out", "w", stdout);

    scanf ("%d", &n);

    int k = 0, nr = 0;
    for (int i = 1; i <= n; ++i)
    {
        int a, x;
        scanf ("%d", &a);

        if (a != 3) scanf ("%d", &x);
        if (a == 1)
        {
            v[++k].f = x;
            vv[++nr] = k;
            v[k].s = nr;
            heapup (k);
        }

        else if (a == 2)
        {
            swap (v[vv[x]], v[k]);
            swap (vv[v[k].s], vv[x]);
            heapdown (vv[x], k - 1);
            --k;
        }

        else if (a == 3) printf ("%d\n", v[1].f);
    }

    return 0;
}