Cod sursa(job #3330164)

Utilizator marelucaMare Luca Ghita mareluca Data 17 decembrie 2025 22:20:51
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.82 kb
#include <bits/stdc++.h>

const std::string FILE_NAME = "apm";
std::ifstream fin(FILE_NAME + ".in");
std::ofstream fout(FILE_NAME + ".out");

const int NMAX = 105;

int n, m, costMinim;
std::vector<int> x, y, c, indice, radacina, rang, apm;

bool cmp(int i, int j) {
    return (c[i] < c[j]);
}

void read() {
    fin >> n >> m;

    x.resize(m);
    y.resize(m);
    c.resize(m);

    indice.resize(m);

    for(int i = 0; i < m; ++i) {
        fin >> x[i] >> y[i] >> c[i];
        --x[i], --y[i];
        indice[i] = i;
    }

    rang.resize(n);
    radacina.resize(n);

    for(int i = 0; i < n; ++i) {
        radacina[i] = i;
        rang[i] = 1;
    }
}

int find_R(int node) {
    int R;

    // Cautarea radacinii
    for(R = node;
        radacina[R] != R;
        R = radacina[R]);

    // Compresia drumului
    while(node != radacina[node]) {
        int r = radacina[node];
        radacina[node] = R;
        node = r;
    }

    return R;
}

void unite(int node_A, int node_B) {
    if(node_A == node_B) return;

    if(rang[node_A] > rang[node_B])
        radacina[node_B] = radacina[node_A];
    else
        radacina[node_A] = radacina[node_B];

    if(rang[node_A] == rang[node_B]) ++rang[node_B];
}

void kurskal() {
    std::sort(indice.begin(), indice.end(), cmp);

    for(int i = 0; i < m; ++i) {
        if(find_R(x[indice[i]]) != find_R(y[indice[i]])) {
            unite(find_R(x[indice[i]]), find_R(y[indice[i]]));
            apm.push_back(indice[i]);
            costMinim += c[indice[i]];
        }
    }
}

void write() {
    fout << costMinim << '\n';
    fout << apm.size() << '\n';
    for(int i : apm) {
        fout << x[i] + 1 << ' ' << y[i] + 1 << '\n';
    }
}

int main() {
    std::ios_base::sync_with_stdio(false);
    fin.tie(nullptr);
    fout.tie(nullptr);

    read();
    kurskal();
    write();

    return 0;
}