Cod sursa(job #2978991)

Utilizator proflaurianPanaete Adrian proflaurian Data 14 februarie 2023 18:23:39
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.92 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
const int N = 100010;
int n,m,costArbore,cst,x,y,T[N];
vector<tuple<int,int,int>> M;
vector<pair<int,int>> A;
int getRoot(int nod)
{
    if(T[nod]==nod)
        return nod;
    T[nod]=getRoot(T[nod]);
    return T[nod];
}
int main()
{
    f>>n>>m;
    for(int i=1;i<=n;i++)
        T[i]=i;
    for(int i=1;i<=m;i++)
    {
        f>>x>>y>>cst;
        M.push_back(make_tuple(cst,x,y));
    }
    sort(M.begin(),M.end());
    for(auto it:M)
    {
        tie(cst,x,y)=it;
        int rx=getRoot(x);
        int ry=getRoot(y);
        if(rx!=ry)
        {
            costArbore+=cst;
            A.push_back(make_pair(x,y));
            T[rx]=ry;
        }
    }
    g<<costArbore<<'\n';
    g<<A.size()<<'\n';
    for(auto it:A)
    {
        tie(x,y)=it;
        g<<x<<' '<<y<<'\n';
    }
    return 0;
}