Cod sursa(job #2017510)

Utilizator Claudiu07Pana Claudiu Claudiu07 Data 1 septembrie 2017 15:16:36
Problema Arbore partial de cost minim Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.01 kb
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
struct muchie
{
    int x,y,c;
};
muchie a[400001];
int n,h[200001],sol[200001][2],m,s,nr,tata[200001];
bool cmp( const muchie &a, const muchie &b)
{
    if(a.c<b.c) return 1;
    return 0;
}
int radacina(int x)
{
    if(tata[x]==x) return x;
     return radacina(tata[x]);
}
int main()
{
    int i,j;
    f>>n>>m;
    for(i=1; i<=n; i++)
    {
        h[i]=i;
       tata[i]=i;
    }
    for(i=1; i<=m; i++)
    {
        f>>a[i].x>>a[i].y>>a[i].c;
    }
    sort(a+1,a+m+1,cmp);
    i=1,j=0;
    while(j<n-1)
    {
        if(radacina(a[i].x)!=radacina(a[i].y))
        {
            s=s+a[i].c;
            j++;
            sol[j][0]=a[i].x;
            sol[j][1]=a[i].y;
            tata[radacina(a[i].y)]=radacina(a[i].x);
        }
        i++;
    }
    g<<s<<'\n';
    g<<j<<'\n';
    for(i=1; i<=j; i++)
        g<<sol[i][0]<<' '<<sol[i][1]<<'\n';
}