Cod sursa(job #1846455)

Utilizator tziplea_stefanTiplea Stefan tziplea_stefan Data 12 ianuarie 2017 20:57:09
Problema Heapuri Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.61 kb
#include <fstream>
#define VAL 200005

using namespace std;

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

int N, M, i, j;
int op, x, a, b;
int v[VAL];
int poz[VAL];
int heap[VAL];

void percolate(int p, int x)
{
    while (heap[p]<heap[p / 2] && p!=1)
    {
        swap(heap[p], heap[p / 2]);
        swap(poz[p], poz[p / 2]);
        p/=2;
    }
}

void sift(int p)
{
    while (p<N)
    {
        fout << " XXX " << p << " XXX\n";
        a=2*p;
        b=2*p+1;
        if (a<=N && b<=N)
        {
            if (heap[a]<=heap[b] && heap[p]>heap[a])
            {
                swap(heap[p], heap[a]);
                swap(poz[p], poz[a]);
                p=a;
            }
            if (heap[b]<=heap[a] && heap[p]>heap[b])
            {
                swap(heap[p], heap[b]);
                swap(poz[p], poz[b]);
                p=b;
            }
        }
        else
        {
            if (a<=N && heap[p]>heap[a])
            {
                swap(heap[p], heap[a]);
                swap(poz[p], poz[a]);
                p=a;
            }
            if (a>N)
              break;
        }
    }
}

int main()
{
    fin >> M;
    for (i=1; i<=M; i++)
    {
        fin >> op;
        if (op==3)
          fout << heap[1] << '\n';
        else
          fin >> x;
        if (op==1)
        {
            v[++N]=x;
            poz[N]=N;
            heap[N]=x;
            percolate(N, x);
        }
        if (op==2)
        {
            swap(heap[poz[x]], heap[N]);
            swap(poz[x], poz[N]);
            N--;
            sift(poz[N+1]);
        }
    }
    fin.close();
    fout.close();
    return 0;
}