Cod sursa(job #2155893)

Utilizator stefii_predaStefania Preda stefii_preda Data 8 martie 2018 11:26:52
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.37 kb
#include <fstream>
#include <algorithm>

using namespace std;

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

const int N = 200005;

int tata[N], niv[N];

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

muchie m[2*N], sol[N];

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

void Union(int r1, int r2)
{
    if(niv[r1] < niv[r2])
        tata[r1] = r2;
    else
        if(niv[r1] > niv[r2])
        tata[r2] = r1;
    else
    {
        tata[r1] = r2;
        niv[r2] ++;
    }
}

int Find(int x)
{
    int y, rad = x;
    while(tata[rad] != rad)
        rad = tata[rad];
    while(tata[x] != x)
    {
        y = tata[x];
        tata[x] = rad;
        x = y;
    }
    return rad;
}

int main()
{
    int n, edg, i;
    in >> n >> edg;
    for(i = 1; i <= edg; i++)
    {
        in >> m[i].x >> m[i].y >> m[i].c;
    }
    sort(&m[1], &m[edg + 1], cmp);
    for(i = 1; i <= n; i++)
        tata[i] = i;
    int rad1, rad2, k = 0, cost = 0;
    for(i = 1; i <= edg; i++)
    {
        rad1 = Find(m[i].x);
        rad2 = Find(m[i].y);
        if(rad1 != rad2)
        {
            k++;
            sol[k] = m[i];
            cost += m[i].c;
            Union(rad1, rad2);
        }
    }
    out << cost <<"\n" << k <<"\n";
    for(i = 1; i <= k; i++)
        out << sol[i].x <<" " << sol[i].y <<"\n";
    return 0;
}