Cod sursa(job #2950767)

Utilizator pifaDumitru Andrei Denis pifa Data 4 decembrie 2022 17:16:24
Problema Arbore partial de cost minim Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <bits/stdc++.h>

using namespace std;

ifstream in("apm.in");
ofstream out("apm.out");

const int N = 2e5;

struct muchie
{
    int x, y, z, viz;
};

muchie v[2 * N + 1];

int parent[N + 1];

int n,m;

int compar(muchie a, muchie b)
{
    return a.z < b.z;
}

int Find(int nod)
{
    if(nod == parent[nod])
        return nod;
    return parent[nod] = Find(parent[nod]);
}

int main()
{
    in >> n >> m;
    for(int i = 1; i <= n; i++)
        parent[i] = i;
    for(int i = 1; i <= m; i++)
    {
        in >> v[i].x >> v[i].y >> v[i].z;
    }
    sort(v + 1, v + m + 1, compar);
    int sum = 0, cnt = 0;
    for(int i = 1; i <= m; i++)
    {
        int a = Find(v[i].x), b = Find(v[i].y);
        if(a != b)
        {
            parent[b] = a;
            sum += v[i].z;
            cnt++;
            v[i].viz = 1;
        }
    }
    out << sum << '\n' << cnt << '\n';
    for(int i = 1; i <= n; i++)
    {
        if(v[i].viz)
            out << v[i].x << ' ' << v[i].y << '\n';
    }
    return 0;
}