Cod sursa(job #2507540)

Utilizator Horia20Bistriceanu Horia Horia20 Data 10 decembrie 2019 13:50:59
Problema Arbore partial de cost minim Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include<fstream>
#include<algorithm>
using namespace std;
int n,m,conex[20005],cnt;
long long costmin;
struct muchie{
int x,y,c;
};
muchie v[20005],sol[20005];
void citire()
{
    ifstream f("apm.in");
    f>>n>>m;
    for(int i=1;i<=m;++i)
        f>>v[i].x>>v[i].y>>v[i].c;
    for(int i=1;i<=n;++i)
        conex[i]=i;
}
int test(muchie a, muchie b)
{
    return a.c<b.c;
}
void kruskal()
{
    sort(v+1,v+m+1,test);
    int k=1;
    while(cnt<n-1)
    {
        if(conex[v[k].x]!=conex[v[k].y])
        {
            int w1=conex[v[k].x];
            int w2=conex[v[k].y];
            sol[++cnt]=v[k];
            costmin+=v[k].c;
            for(int i=1;i<=n;++i)
                if(conex[i]==w1)
                  conex[i]=w2;
        }
        ++k;
    }
}
int main()
{
    citire();
    kruskal();
    ofstream g("apm.out");
    g<<costmin<<'\n';
    g<<n-1<<'\n';
    for(int i=1;i<n;++i)
        g<<sol[i].x<<' '<<sol[i].y<<'\n';
    g.close();
    return 0;
}