Cod sursa(job #1012161)

Utilizator Dddarius95Darius-Florentin Neatu Dddarius95 Data 18 octombrie 2013 13:03:19
Problema Arbore partial de cost minim Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 0.87 kb
#include <fstream>
#include <algorithm>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");

struct nod{ int x, y, c; } v[400005];
int cmp(nod a, nod b) { return a.c<b.c; }
int N, M, sol[200005], K, GR[200005], cost;

int grupa(int nod)
{
    if (GR[nod]==nod) return nod;
    return grupa(GR[nod]);
}

void uneste(int x, int y)
{
    GR[grupa(x)]=grupa(y);
}

int main()
{
    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, cmp);
    for (int i=1; i<=N; ++i) GR[i]=i;

    for (int i=1; i<=M && K!=N-1; ++i)
    {
        if ( grupa(v[i].x) != grupa(v[i].y) )
        {
            cost+=v[i].c; uneste(v[i].x, v[i].y);
            sol[++K]=i;
        }
    }
    g<<cost<<'\n'<<K<<'\n';
    for (int i=1; i<N; ++i)
        g<<v[sol[i]].x<<' '<<v[sol[i]].y<<'\n';

    return 0;
}