Cod sursa(job #3201132)

Utilizator AlbertPavPavalache Albert AlbertPav Data 6 februarie 2024 20:53:02
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
const int NMAX = 200001, MMAX = 400001;

struct muchie
{
    int i, j, c;
};

int n, m, nm, cost, CC[NMAX], NrM[NMAX];
muchie M[MMAX];

ifstream f("apm.in");
ofstream g("apm.out");

inline bool comp(muchie &a,muchie &b)
{
    return a.c<b.c;
}

void citire()
{
    f>>n>>m;
    for(int i=1;i<=m;i++)
        f>>M[i].i>>M[i].j>>M[i].c;
}

int Find(int k)
{
    if(CC[k]==0)
        return k;
    return CC[k]=Find(CC[k]);
}

void Kruskal()
{
    int ci,cj;
    sort(M+1,M+m+1,comp);
    for(int i=1;i<=m;i++)
    {
        ci=Find(M[i].i);
        cj=Find(M[i].j);
        if(ci!=cj)
        {
            cost+=M[i].c;
            CC[ci]=cj;
            NrM[++nm]=i;
            if(nm==n-1)
                break;
        }
    }
}

void afis()
{
    g<<cost<<'\n';
    g<<nm<<'\n';
    for(int i=1;i<=nm;i++)
        g<<M[NrM[i]].i<<' '<<M[NrM[i]].j<<'\n';
}

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