Cod sursa(job #2181793)

Utilizator cristibogdanPatrascu Cristian cristibogdan Data 21 martie 2018 20:53:43
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1 kb
#include <fstream>
#include <algorithm>
#define NMAX 200001
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
int n,m,i,tata[NMAX],s,nr;
struct du{
    int x,y,cost;}muchii[2*NMAX];
struct du2{
    int x,y;}sol[2*NMAX];
int cmp(du a, du b){
    return a.cost<b.cost;}
int root(int nod){
    if(tata[nod]==nod)
        return nod;
    tata[nod]=root(tata[nod]);
    return tata[nod];
    }

int main()
{
    f>>n>>m;
    for(i=1;i<=m;i++)
        f>>muchii[i].x>>muchii[i].y>>muchii[i].cost;
        for(int i = 1;i <=n; ++i)
            tata[i] = i;
    sort(muchii+1,muchii+m+1,cmp);
    for(i=1;i<=m;i++)
        if(root(muchii[i].x)!=root(muchii[i].y)){
            tata[root(muchii[i].x)]=root(muchii[i].y);

                s+=muchii[i].cost;
                sol[++nr].x=muchii[i].x;
                sol[nr].y=muchii[i].y;
            }

        g<<s<<'\n'<<n-1<<'\n';
        for(i=1;i<=n-1;i++)
            g<<sol[i].x<<" "<<sol[i].y<<'\n';
    return 0;
}