Cod sursa(job #1840176)

Utilizator miha1000Dica Mihai miha1000 Data 3 ianuarie 2017 21:04:06
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.18 kb
#include <iostream>
#include <fstream>
#include <algorithm>

using namespace std;

struct muchie{
    int x,y,c;
};

muchie v[400000];




int p[200000];

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

int findSet(int x){
   if (p[x]==x){
        return x;
   }
   return p[x]=findSet(p[x]);
}

bool insameSet(int x, int y){
    return (findSet(x)==findSet(y));
}

void unionSet(int x,int y){
    if (!insameSet(x,y)){
        int a=findSet(x);
        int b=findSet(y);
        p[a]=b;
    }
}

int main()
{
    int n,m,i,j,x,y,lungime;
    ifstream f("apm.in");
    ofstream g("apm.out");
    f >> n >> m;
    lungime=0;
    for (i=1;i<=m;i++){
        f >> v[i].x >> v[i].y>> v[i].c;
    }
    for(i=1;i<=n;i++){
        p[i]=i;
    }
    n=0;
    sort(v+1,v+m+1,comp);
    for(i=1;i<=m;i++){
        x=v[i].x;
        y=v[i].y;
        if (!insameSet(x,y)){
            n++;
            unionSet(x,y);
            lungime=lungime+v[i].c;
            v[n].x=x;
            v[n].y=y;
        }
    }
    g << lungime << "\n";
    g << n << "\n";
    for (i=1;i<=n;i++){
        g << v[i].x << " " << v[i].y << "\n";

    }
}