Cod sursa(job #2932637)

Utilizator dragosteleagaDragos Teleaga dragosteleaga Data 3 noiembrie 2022 14:44:27
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
#include <algorithm>
#include <fstream>
#include <iostream>
using namespace std;
ifstream f("amp.in");
ofstream o("amp.out");
const int NMax = 400005;
pair<int, int> P[NMax];
int N, M, Total, TT[NMax], k, RG[NMax];

struct Edge {
    int x, y, c;
} V[NMax];
bool Compare(Edge a, Edge b) {
    return a.c < b.c;
}
void Read() {
    f >> N >> M;
    for (int i = 1; i <= M; i++)
        f >> V[i].x >> V[i].y >> V[i].c;
    sort(V + 1, V + M + 1, Compare);
}
int Find(int nod) {
    while (TT[nod] != nod)
        nod = TT[nod];
    return nod;
}
void Unite(int x, int y) {
    if (RG[x] < RG[y])
        TT[x] = y;
    if (RG[y] < RG[x])
        TT[y] = x;
    if (RG[x] == RG[y]) {
        TT[x] = y;
        RG[y]++;
    }
}
void Solve() {
    for (int i = 1; i <= M; i++) {
        //cout << "Incerc " << V[i].x << " cu " << V[i].y << "\n";
        if (Find(V[i].x) != Find(V[i].y)) {
            //cout << "Unesc " << V[i].x << " cu " << V[i].y << "\n\n";
            Unite(Find(V[i].x), Find(V[i].y));
            P[++k].first = V[i].x;
            P[k].second = V[i].y;
            Total += V[i].c;
        }
    }
}
int main() {
    Read();
    for(int i=1;i<=M;i++){
        TT[i]=i;
        RG[i]=1;
    }
    Solve();
    o<<Total<<"\n";
    o<<N-1<<"\n";
    for(int i = 1; i <= k; i++)
        o << P[i].second << " " << P[i].first << "\n";
    return 0;
}