Cod sursa(job #1918739)

Utilizator paunmatei7FMI Paun Matei paunmatei7 Data 9 martie 2017 16:42:51
Problema Arbore partial de cost minim Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.35 kb
#include <fstream>
#include <algorithm>
#include <vector>

using namespace std;

const int NMAX = 200007;
const int MMAX = 400007;

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

struct apm {
    int x;
    int y;
    int c;
};

apm a[MMAX];
vector < apm > sol;
int n, m;
int T[NMAX], H[NMAX];

int father(int node) {
    if(node == T[node]) {
        return node;
    }
    return father(T[node]);
}

void unite(int a, int b) {
    if(H[a] > H[b]) {
        T[b] = a;
    }
    else {
        if(H[b] < H[a]) {
            T[a] = b;
        }
        else{
            ++H[a];
            T[b] = a;
        }
    }
}

int cmp(apm a, apm b) {
    return a.c < b.c;
}

int main() {
    cin >> n >> m;
    for(int i = 1; i <= n; ++i) {
        T[i] = i;
        H[i] = 1;
    }
    for(int i = 1; i <= m; ++i) {
        cin >> a[i].x >> a[i].y >> a[i].c;
    }
    sort(a + 1, a + m + 1, cmp);
    int ans = 0;
    for(int i = 1; i <= m; ++i) {
        int Tx = father(a[i].x);
        int Ty = father(a[i].y);
        if(Tx != Ty) {
            ans += a[i].c;
            unite(Tx, Ty);
            sol.push_back(a[i]);
        }
    }
    cout << ans << "\n" << sol.size() << "\n";
    for(int i = 0; i < sol.size(); ++i) {
        cout << sol[i].x << " " << sol[i].y << "\n";
    }
    return 0;
}