Cod sursa(job #3173245)

Utilizator elisa.ipateElisa Ipate elisa.ipate Data 22 noiembrie 2023 10:10:19
Problema Arbore partial de cost minim Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.09 kb
#include <iostream>
#include <fstream>
#include <algorithm>


#define mmax 400000
#define nmax 200000


using namespace std;

struct info{
    int cost, x, y;
};

info edges[mmax];
int v[nmax], sol[nmax];

bool sortare( info &a, info &b ) {
    if( a.cost <= b.cost )
        return true;
    return false;
}

int sef( int a ) {
    int aux;
    if( v[a] == a )
        return a;
    aux = sef( v[a] );
    v[a] = aux;
    return aux;
}

int main()
{
    int n, m, i, a, b, s = 0, j;
    ifstream cin("apm.in");
    ofstream cout("apm.out");
    cin >> n >> m;
    for( i = 0; i < m; i++ )
        cin >> edges[i].x >> edges[i].y >> edges[i].cost;

    sort( edges, edges + m, sortare );

    for( i = 1; i <= n; i++ )
        v[i] = i;

    j = 0;
    for( i = 0; i < m; i++ ) {
        a = sef(edges[i].x);
        b = sef(edges[i].y);
        if( a != b ) {
            v[a] = b;
            sol[j] = i;
            j++;
            s += edges[i].cost;
        }
    }
    cout << s << "\n" << j << "\n";
    for( i = 0; i < j; i++ )
        cout << edges[sol[i]].x << " " << edges[sol[i]].y << "\n";

    return 0;
}