Cod sursa(job #2166612)

Utilizator razvanlgu31Razvan Lungu razvanlgu31 Data 13 martie 2018 17:59:05
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.5 kb
#include <fstream>
#include <algorithm>
#include <vector>
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
int i, n, m, tata[200010], sum;
struct muchie
{
    int d, x, y;
}mu[400010];
vector < pair < int, int > > sol;
int tat(int a)
{
    while(tata[a] > 0)
        a = tata[a];
    return a;
}
void kruskal()
{
    int nr = 0;
    for(int i = 1; i <= m; i++)
    {
        int a, b;
        a = tat(mu[i].x);
        b = tat(mu[i].y);
        if(a != b)
        {
//                fout<< mu[i].d<<" "<<mu[i].x<<" "<<mu[i].y<<'\n';
            sum += mu[i].d;
            nr++;
            if(tata[a] > tata[b])
            {
                tata[b] += tata[a];
                tata[a] = b;
            }
            else
            {
                tata[a] += tata[b];
                tata[b] = a;
            }
            sol.push_back({mu[i].x, mu[i].y});
        }
        if(nr == n-1)
    return;
    }

}
int cmp(muchie a, muchie b)
{
    return a.d < b.d;
}
int main()
{
    fin >> n >> m;
    for(i = 1; i <= m; i++)
    {
        fin >> mu[i].x >> mu[i].y >> mu[i].d;
    }
    for(i = 1; i <= n; i++)
        tata[i] = -1;
    sort(mu + 1, mu + m + 1, cmp);
//    for(i=1;i<=m;i++)
//        fout<<mu[i].d<<" " <<mu[i].x<<" "<<mu[i].y<<'\n';
    kruskal();
    fout << sum << '\n' << n-1 << '\n';
    for(i = 0; i < sol.size(); i++)
    {
        fout << sol[i].first << " " << sol[i].second << '\n';
    }
    return 0;
}