Cod sursa(job #1884766)

Utilizator CriistinaMicula Cristina Criistina Data 19 februarie 2017 10:53:08
Problema Heapuri Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.53 kb
#include <iostream>
#include <fstream>
#define Nmax 200001

using namespace std;
ifstream f("heapuri.in");
ofstream g("heapuri.out");

int n, v[Nmax]; ///v[x]=ce element se afla in heap pe poz x
int ord[Nmax]; ///ord[x]=al catelea a fost introdus elementul aflat pe poz x in heap
int poz[Nmax]; ///poz[x]=pe ce poz din heap se afla elem introdus a x-a oara
int nr, s;

void inserare(int x, int k) ///inserare elementul x pe poz k
{
    v[k]=x;
    ord[k]=s;
    poz[ord[k]]=k;
    while(k>1 && v[k]<v[k/2])
    {
        swap(v[k], v[k/2]);
        poz[ord[k]]=k/2;
        poz[ord[k/2]]=k;
        swap(ord[k], ord[k/2]);
        k=k/2;
    }
}
void stergere(int k) ///stergere elementul de pe pozitia k din heap
{
    v[k]=v[nr];
    poz[ord[nr]]=k;
    ord[k]=ord[nr];
    nr--;
    while(k<=nr/2 && (v[k]>v[2*k] || (2*k+1<=nr && v[k]>v[2*k+1])))
    {
        int ind;
        if(2*k+1>nr || v[2*k]<v[2*k+1])
            ind=2*k;
        else
            ind=2*k+1;
        swap(v[k], v[ind]);
        poz[ord[k]]=ind;
        poz[ord[ind]]=k;
        swap(ord[k], ord[ind]);
        k=ind;
    }
}
int main()
{
    f>>n;
    for(int i=1;i<=n;i++)
    {
        int x;
        f>>x;
        if(x==1)
        {
            f>>x;
            nr++;
            s++;
            inserare(x, nr);
        }
        else if(x==2)
        {
            f>>x;
            stergere(poz[x]);  ///stergere elementul introdus a x-a oara in heap
        }
        else
            g<<v[1]<<"\n";
    }
}