Cod sursa(job #1097272)

Utilizator rares96cheseliRares Cheseli rares96cheseli Data 3 februarie 2014 11:42:53
Problema Arbore partial de cost minim Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.98 kb
#include <fstream>
#include <algorithm>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");

struct muchie{ int x, y, c; }m[400002];
struct cmp
{
    bool operator()(const muchie &a, const muchie &b)const
    {
        if (a.c<b.c) return 1;
        return 0;
    }
};
int N, M, K, sol[400002], GR[200002], cost;

int grupa(int nod)
{
    if (GR[nod]==nod) return nod;
    GR[nod]=grupa(GR[nod]);
    return 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>>m[i].x>>m[i].y>>m[i].c;

    sort(m+1, m+N+1, cmp());
    for (int i=1; i<=N; ++i)
        GR[i]=i;

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

    g<<cost<<' '<<K<<'\n';
    for (int i=1; i<=K; ++i)
        g<<m[sol[i]].x<<' '<<m[sol[i]].y<<'\n';
    return 0;
}