Cod sursa(job #2862615)

Utilizator Seress26Seres Artur Seress26 Data 5 martie 2022 16:34:48
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#define NMAX 400001
///KRUSKAL.
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
int n,m,k,Total,TT[NMAX],RG[NMAX];
pair <int,int> P[NMAX];
struct Muchii
{
    int x;
    int y;
    int c;
}U[NMAX];
bool Compara(Muchii a, Muchii b)
{
    return a.c < b.c;
}
void Citire()
{
    f>>n;
    f>>m;
    for(int i=1;i<=m;i++)
    {
        f>>U[i].x;
        f>>U[i].y;
        f>>U[i].c;
    }
}
int Gasit(int Nod)
{
    while(TT[Nod]!=Nod)
        Nod=TT[Nod];
    return Nod;
}
void Unire(int x,int y)
{
    if(RG[x]<RG[y])
        TT[x]=y;
    if(RG[x]>RG[y])
        TT[y]=x;
    if(RG[x]==RG[y])
    {
        TT[x]=y;
        RG[y]++;
    }
}
void Rezolvare()
{
    for(int i=1;i<=m;i++)
    {
        if(Gasit(U[i].x)!=Gasit(U[i].y))
        {
            Unire(Gasit(U[i].x),Gasit(U[i].y));
            k++;
            P[k].first=U[i].x;
            P[k].second=U[i].y;
            Total=Total+U[i].c;
        }
    }
}
void Afisare()
{
    g<<Total<<'\n';
    g<<k<<'\n';
    for(int i=1;i<=k;i++)
        g<<P[i].second<<' '<<P[i].first<<'\n';
}
int main()
{
    Citire();
    for(int i=1;i<=m;i++)
    {
        TT[i]=i;
        RG[i]=1;
    }
    sort(U+1,U+m+1,Compara);
    Rezolvare();
    Afisare();

    return 0;
}