Cod sursa(job #2047834)

Utilizator eusebiu_gageaGagea Eusebiu-Andrei eusebiu_gagea Data 25 octombrie 2017 13:58:29
Problema Heapuri Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.64 kb
#include <stdio.h>
#include <assert.h>

#define maxn 200010

int N, L, NR;
int A[maxn], Heap[maxn], Pos[maxn];

void push(int x)
{
    int aux;

    while (x/2 && A[Heap[x]]<A[Heap[x/2]])
    {
        aux = Heap[x];
        Heap[x] = Heap[x/2];
        Heap[x/2] = aux;

        Pos[Heap[x]] = x;
        Pos[Heap[x/2]] = x/2;
        x /= 2;
    }
}

void pop(int x)
{
    int aux, y = 0;

    while (x != y)
    {
        y = x;

        if (y*2<=L && A[Heap[x]]>A[Heap[y*2]]) x = y*2;
        if (y*2+1<=L && A[Heap[x]]>A[Heap[y*2+1]]) x = y*2+1;

        aux = Heap[x];
        Heap[x] = Heap[y];
        Heap[y] = aux;

        Pos[Heap[x]] = x;
        Pos[Heap[y]] = y;
    }
}

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

    int i, x, cod;

    scanf("%d ", &N);

    assert(1<=N && N<=200000);

    for (i=1; i<=N; i++)
    {
        scanf("%d ", &cod);
        assert(1<=cod && cod<=3);
        if (cod < 3)
        {
            scanf("%d ", &x);
            assert(1<=x && x<=1000000000);
        }

        if (cod == 1)
        {
            L++, NR++;
            A[NR] = x;
            Heap[L] = NR;
            Pos[NR] = L;

            push(L);
        }

        if (cod == 2)
        {
            A[x] = -1;
            assert(Pos[x] != 0);
            assert(1<=x && x<=NR);
            push(Pos[x]);

            Pos[Heap[1]] = 0;
            Heap[1] = Heap[L--];
            Pos[Heap[1]] = 1;
            if (L>1) pop(1);
        }

        if (cod == 3) printf("%d\n", A[Heap[1]]);
    }

    return 0;
}