Cod sursa(job #1298868)

Utilizator witselWitsel Andrei witsel Data 23 decembrie 2014 11:55:00
Problema Heapuri Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.35 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("heapuri.in");
ofstream fout("heapuri.out");
const int NMAX=200000;
int H[NMAX],v[NMAX],n,N,i,m,op,j,x;
void inserare(int);
void percolate(int);
void stergere(int);
void shift(int );
int main()
{
    fin>>N;
    while(N--)
    {
        fin>>op;
        switch(op)
        {
            case 1:fin>>x;
                    n++;
                    v[++j]=x;
                    inserare(x);
                    break;
            case 2:fin>>x; stergere(v[x]);
            break;
            case 3:fout<<H[1]<<"\n";
        }
    }
}

void inserare(int x)
{
    H[n]=x;
    percolate(n);
}
void percolate(int k)
{
    while(k>1 && H[k]<H[k/2])
    {
        swap(H[k],H[k/2]);
        k=k/2;
    }
}
 void stergere(int val)
 {
     for(j=1;j<=n && H[j]!=val;j++);
     swap(H[j],H[n]);
     n--;
     if(H[j]<H[j/2])
        percolate(j);
     else shift(j);
 }
 void shift(int k)
 {
     int ok=1;
     while(k<=n/2 && ok==1)
     {
         ok=0;
         if(k*2+1<=n && H[k*2+1]<=H[k*2] && H[k]>H[k*2+1])
         {
             swap(H[k*2+1],H[k]);
             k=k*2+1;
             ok=1;
         }
         else if(H[k]>H[2*k])
         {
             swap(H[k],H[2*k]);
             k=2*k;
             ok=1;
         }
     }
 }