Cod sursa(job #2640764)

Utilizator Iustin01Isciuc Iustin - Constantin Iustin01 Data 8 august 2020 09:02:15
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.1 kb
#include <bits/stdc++.h>
#define MAX 200005
using namespace std;
ifstream in("apm.in");
ofstream out("apm.out");

struct ex{
    int x , y, cost;
}g[MAX], sol[MAX];

int n, m, rang[MAX], z, t[MAX];
long long int sum;

bool cmp(ex a, ex b){
    return a.cost < b.cost;
}

int radacina(int k){
    if(!t[k])
        return k;
    else
        return t[k] = radacina(t[k]);
}

void unire(int rp, int rk){
        if(rang[rk] > rang[rp])
            t[rp] = rk;
        else{
            t[rk] = rp;
            if(rang[rk] == rang[rp])
                rang[rp]++;
        }
}

int main(){
    in>>n>>m;
    for(int i = 1; i <= m; i++)
        in>>g[i].x>>g[i].y>>g[i].cost;
    sort(g + 1, g + m + 1, cmp);

    for(int i = 1; i <= m; i++){
        int rk = radacina(g[i].x), rp = radacina(g[i].y);
        if(rk != rp){
            unire(rp, rk);
            sol[++z].x = g[i].x, sol[z].y = g[i].y, sol[z].cost = g[i].cost;
            sum += g[i].cost;
        }
    }
    out<<sum<<"\n"<<z<<"\n";
    for(int i = 1; i <= z; i++)
        out<<sol[i].x<<" "<<sol[i].y<<"\n";
}