Cod sursa(job #3136530)

Utilizator BetJohn000Ioan Benescu BetJohn000 Data 6 iunie 2023 20:45:50
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.39 kb
#include <fstream>
#include <iostream>
#include <algorithm>

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

pair<int,int> Perechi[400005];
int k;
int N,M,Total,Tata[200001],Rang[400005];
struct Muchie{
    int x,y,cost;
}V[200001];
bool Compare(Muchie a, Muchie b)
{
    return a.cost < b.cost;
}
void Read()
{
    f>>N>>M;
    for(int i=1;i<=M;i++)
        f>>V[i].x>>V[i].y>>V[i].cost;
    sort(V+1,V+M+1, Compare);
    for(int i=1;i<=N;i++)
    {
        Tata[i] = i;
        Rang[i] = 1;
    }
}
int Find(int Nod)
{
    while(Tata[Nod]!=Nod)
        Nod = Tata[Nod];
    return Nod;
}
void Unire(int x, int y)
{
    if(Rang[x]<Rang[y])
        Tata[x] = y;
    if(Rang[x] > Rang[y]) 
        Tata[y] = x;
    if(Rang[x] == Rang[y])
    {
        Tata[x] = y;
        Rang[y]++;
    }
}
void Afisare()
{
    g<<Total<<"\n";
    g<<N-1<<"\n";
    for(auto pereche:Perechi)
        g<<pereche.first<<" "<<pereche.second<<"\n";
}
void Rezolva()
{
    for(int i = 1;i<=M;i++)
    {
        int tatal_x = Find(V[i].x);
        int tatal_y = Find(V[i].y);
        if(tatal_x != tatal_y)//tatii nu sunt egali, adica nu formeaza ciclu
        {
            Perechi[++k].first = V[i].x;
            Perechi[k].second = V[i].y;
            Total +=V[i].cost;
        }
    }
}
int main()
{
    Read();
    Rezolva();
    Afisare();
}