Cod sursa(job #3215342)

Utilizator cristiana_6Dode Cristiana cristiana_6 Data 14 martie 2024 20:35:46
Problema Arbore partial de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.09 kb
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;

ifstream fin("apm.in");
ofstream fout("apm.out");

int n, m, cc[200001], s, nr, g[200000], k, rg[400001];

struct muchie{
    int x, y, c;
}v[400001];

bool verif(muchie a, muchie b){
    return a.c < b.c;
}

void afis(){
    fout<<s<<'\n'<<nr<<'\n';
    for(int i=1; i<=nr; i++)
        fout<<v[g[i]].x<<' '<<v[g[i]].y<<'\n';
}

int grupa(int i){
    while(cc[i] != i)
        i=cc[i];
    return i;
}

void unire(int x, int y){
    if(rg[x] < rg[y]) cc[y]=x;
    else 
    if(rg[x] > rg[y]) cc[x]=y;
    else{
        cc[x]=y;
        rg[y]++;
    }
}

int main()
{
    fin>>n>>m;
    for(int i=1; i<=m; i++){
        fin>>v[i].x>>v[i].y>>v[i].c;
    }
    sort(v+1, v+m+1, verif);
    for(int i=1; i<=n; i++)
        cc[i]=i;
    for(int i=1; i<=m; i++){
        if(grupa(v[i].x) != grupa(v[i].y)){
            int x1=grupa(v[i].x), y1=grupa(v[i].y);
            unire(x1, y1);
            g[++nr]=i;
            s+=v[i].c;
        }
    }
    afis();
    return 0;
}