Cod sursa(job #1577161)

Utilizator Arodoet96Teodora Stoleru Arodoet96 Data 23 ianuarie 2016 11:54:11
Problema Heapuri Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.86 kb
#include <fstream>
#define DMAX 2000004

using namespace std;

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

int N;
int H[DMAX], n;
int ord[DMAX], poz[DMAX];

void inserare(int inf);
void combinare(int i);
int extragere_x(int x);
void citire();
void afisare();

int main()
{
    citire();
    return 0;
}

void citire()
{
    int i, op, x;
    fin>>N;

    for(i=1;i<=N;++i)
    {
        fin>>op;

        //fout<<op<<'\n';

        if(op==1)//se insereaza elementul x in multime
        {
            fin>>x;
            inserare(x);
            //afisare();
        }

        if(op==2)//se sterge elementul intrat al x-lea in multime, in ordine cronologica
        {
            fin>>x;
            extragere_x(poz[x]);
        }

        if(op==3)//se afiseaza elementul minim din multime
            fout<<H[1]<<'\n';

        //fout<<'\n';
    }
}

void inserare(int inf)
{
    int fiu=++n;
    int tata=fiu/2;

    while(tata && H[tata]>inf)
    {
        H[fiu]=H[tata]; poz[ord[tata]]=fiu;
        fiu=tata; tata=tata/2;
    }

    H[fiu]=inf; ord[fiu]=n; poz[n]=fiu;
}

void combinare(int i)
{
    int inf=H[i], tata=i, fiu=2*i;

    while(fiu<=n)
    {
        if(fiu+1<=n && H[fiu+1]<H[fiu])
            ++fiu;//cel mai mic dintre fii

        if(H[fiu]<inf)
        {
            H[tata]=H[fiu];//il promovez pe cel mai mic dintre fii
            poz[ord[fiu]]=tata;

            tata=fiu;
            fiu=2*tata;
        }
            else break;//am gasit locul
    }

    H[tata]=inf; ord[tata]=ord[inf]; poz[ord[inf]]=tata;
}

int extragere_x(int x)
{
    H[x]=H[n--];
    poz[ord[n]]=x;
    combinare(x);
}

void afisare()
{
    int i;
    for(i=1;i<=n;++i)
        fout<<H[i]<<' ';
    fout<<'\n';

    for(i=1;i<=n;++i)
        fout<<poz[i]<<' ';
    fout<<'\n';
}