Cod sursa(job #2620847)

Utilizator CosminBanicaBanica Cosmin CosminBanica Data 29 mai 2020 19:19:30
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.52 kb
#include <algorithm>
#include <fstream>
#include <iostream>
#include <queue>
#include <vector>
#include <tuple>

#define NMAX 200009
#define MMAX 400009
#define kInf (1 << 30)

using namespace std;

typedef tuple<int, int, int> edge;
#define edge_cost(t) std::get<0>(t)
#define first_node(t) std::get<1>(t)
#define second_node(t) std::get<2>(t)

int n, m, cost, p[NMAX];
vector<edge> edges;
vector<edge> apm;

int get_set(int node) {
    if (p[node] != node) {
        p[node] = get_set(p[node]);
    }

    return p[node];
}

void reunion_set(int x, int y) {
    p[get_set(x)] = get_set(y);
}

void Kruskal() {
    for (int i = 1; i <= n; ++i) {
        p[i] = i;
    }

    cost = 0;
    for (auto it = edges.begin(); it != edges.end(); ++it) {
        int x = first_node(*it), y = second_node(*it), c = edge_cost(*it);

        if (get_set(x) != get_set(y)) {
            cost += c;
            apm.push_back(*it);
            reunion_set(x, y);
        }
    }
}

int main() {
    ifstream fin("apm.in");
    ofstream fout("apm.out");
    fin >> n >> m;

    for (int i = 0; i <= m; ++i) {
        int x, y, c;
        fin >> x >> y >> c;

        edges.push_back(edge(c, x, y));
    }

    sort(edges.begin(), edges.end());

    Kruskal();

    fout << cost << "\n";
    fout << (n - 1) << "\n";
    for (auto it = apm.begin(); it != apm.end(); ++it) {
        fout << first_node(*it) << " " << second_node(*it) << "\n";
    }

    fin.close();
    fout.close();
    return 0;
}