Cod sursa(job #3258317)

Utilizator Gabriel_DaescuDaescu Gabriel Florin Gabriel_Daescu Data 21 noiembrie 2024 21:20:03
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.45 kb
#include <fstream>
#include <vector>
#define NMAX 200002
#define INF (1<<30)
using namespace std;
ifstream  fin("apm.in");
ofstream fout("apm.out");
int N,M,H,S,nr,viz[NMAX],heap[NMAX],pozitie[NMAX],tata[NMAX];
vector <int>cost(NMAX,INF);
vector<vector<pair<int,int>>> A(NMAX);

struct solutie
{
    int x,y;
} sol[NMAX];

void citire()
{
    int x,y,c;
    fin>>N>>M;

    for(int i=1; i<=M; i++)
    {
        fin>>x>>y>>c;
        A[x].push_back({y,c});
        A[y].push_back({x,c});
    }
}

void urcare(int poz)
{
    while(poz/2>=1 && cost[heap[poz]]<cost[heap[poz/2]])
    {
        swap(pozitie[heap[poz]],pozitie[heap[poz/2]]);
        swap(heap[poz],heap[poz/2]);
        poz/=2;
    }
}

void coborare(int poz)
{
    while(2*poz<=H)
    {
        int r=2*poz;

        if(r+1<=H && cost[heap[r+1]]<cost[heap[r]])
        {
            r++;
        }

        if(cost[heap[r]]<cost[heap[poz]])
        {
            swap(pozitie[heap[poz]],pozitie[heap[r]]);
            swap(heap[poz],heap[r]);
            poz=r;
        }
        else
        {
            break;
        }
    }
}

void adaugare_heap(int nod)
{
    H++;
    heap[H]=nod;
    pozitie[nod]=H;
    urcare(H);
}

void scoatere_minim()
{
    swap(pozitie[heap[1]],pozitie[heap[H]]);
    swap(heap[1],heap[H]);
    H--;
    coborare(1);
}

int main()
{
    citire();

    nr=0;
    cost[1]=0;
    adaugare_heap(1);

    while(H)
    {
        int nod=heap[1];
        scoatere_minim();

        if(viz[nod])
        {
            continue;
        }

        S+=cost[nod];
        viz[nod]=1;

        if(tata[nod]!=0)
        {
            nr++;
            sol[nr].x=tata[nod];
            sol[nr].y=nod;
        }

        for(int j=0; j<A[nod].size(); j++)
        {
            if(!viz[A[nod][j].first] && cost[A[nod][j].first]>A[nod][j].second)
            {
                cost[A[nod][j].first]=A[nod][j].second;
                tata[A[nod][j].first]=nod;

                if(!pozitie[A[nod][j].first])
                {
                    adaugare_heap(A[nod][j].first);
                }
                else
                {
                    urcare(pozitie[A[nod][j].first]);
                }
            }
        }
    }

    fout<< S << "\n";
    fout<< nr << "\n";
    for(int i=1; i<=nr; i++)
    {
        fout<< sol[i].x << " " << sol[i].y << "\n";
    }

    return 0;
}