Cod sursa(job #3190998)

Utilizator proflaurianPanaete Adrian proflaurian Data 8 ianuarie 2024 16:16:01
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.88 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
/// kruskal
const int N = 100010;
int n,m,c,x,y,rx,ry,T[N],costAPM;
vector<tuple<int,int,int>> E;
vector<pair<int,int>> sol;
int getRoot(int x)
{
    if(T[x]==x)
        return x;
    T[x]=getRoot(T[x]);
    return T[x];
}
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>>c;
        E.push_back(make_tuple(c,x,y));
    }
    sort(E.begin(),E.end());
    for(auto it:E)
    {
        tie(c,x,y)=it;
        rx=getRoot(x);
        ry=getRoot(y);
        while(rx!=ry)
        {
            costAPM+=c;
            sol.push_back(make_pair(x,y));
            T[rx]=ry;
        }
    }
    g<<costAPM<<'\n'<<n-1<<'\n';
    for(auto it:sol)
        g<<it.first<<' '<<it.second<<'\n';
    return 0;
}