Cod sursa(job #2236393)

Utilizator CatincaBCatinca Balinisteanu CatincaB Data 29 august 2018 14:02:40
Problema Arbore partial de cost minim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.94 kb
#include <fstream>
#include <vector>
#include <tuple>
#include <algorithm>

using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
const int N = 100010;
int n,m,i,c,x,y,rx,ry,t[N],cost;
int get_root(int);
vector<tuple<int,int,int>> M;
vector<pair<int,int>> apm;
int main()
{
    f>>n>>m;
    for(;m;m--)
    {
        f>>x>>y>>c;
        M.push_back(make_tuple(c,x,y));
    }
    sort(M.begin(),M.end());
    for(i=1;i<=n;i++)
        t[i]=i;
    for(auto it:M)
    {
        tie(c,x,y)=it;
        rx=get_root(x);
        ry=get_root(y);
        if(rx!=ry)
        {
            t[rx]=ry;
            cost+=c;
            apm.push_back(make_pair(x,y));
        }
    }
    g<<cost<<'\n';
    g<<apm.size()<<'\n';
    for(auto it:apm)
        g<<it.first<<' '<<it.second<<'\n';
    return 0;
}
int get_root(int x)
{
    if(t[x]==x)
        return x;
    t[x]=get_root(t[x]);
    return t[x];
}