Cod sursa(job #1915073)

Utilizator BAC_AndreiBejinariu Andrei Catalin BAC_Andrei Data 8 martie 2017 19:34:48
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.86 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
const int N=200010;
tuple<int,int,int> M[2*N];
pair<int,int> ans[N];
int n,m,c,x,y,k,cst,rx,ry,T[N],root(int);
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;
        M[i]=make_tuple(c,x,y);
    }
    sort(M+1,M+m+1);
    for(int i=1;k<n-1;i++)
    {
        tie(c,x,y)=M[i];
        rx=root(x);
        ry=root(y);
        if(rx!=ry)
        {
            cst+=c;
            ans[++k]={x,y};
            T[rx]=ry;
        }
    }
    g<<cst<<'\n'<<k<<'\n';
    for(int i=1;i<=k;i++)
    {
        tie(x,y)=ans[i];
        g<<x<<' '<<y<<'\n';
    }
    return 0;
}
int root(int nod)
{
    if(T[nod]==nod)
        return nod;
    T[nod]=root(T[nod]);
    return T[nod];
}