Cod sursa(job #1646200)

Utilizator ionut98Bejenariu Ionut Daniel ionut98 Data 10 martie 2016 15:24:11
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.06 kb
#include<fstream>
#include<algorithm>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
int tata[200005],niv[200005],fin[200005],rez,k;
int n,m,i,xx,yy;
struct muchie
{
    int x,y,c;
}a[200005];
bool cmp(muchie nr1,muchie nr2)
{
    return nr1.c<nr2.c;
}
int radacina(int nr)
{
    while(tata[nr]!=nr)
      nr=tata[nr];
    return nr;
}
void add(int xx,int yy)
{
    if(niv[xx]>niv[yy])
      tata[yy]=xx;
    else
      tata[xx]=yy;
    if(niv[xx]==niv[yy])
      niv[yy]++;
}
int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        niv[i]=1;
        tata[i]=i;
    }
    for(i=1;i<=m;i++)
      f>>a[i].x>>a[i].y>>a[i].c;
    rez=0;
    sort(a+1,a+m+1,cmp);
    for(i=1;i<=m;i++)
    {
        xx=radacina(a[i].x);
        yy=radacina(a[i].y);
        if(xx!=yy)
        {
            k++;
            fin[k]=i;
            rez+=a[i].c;
            add(xx,yy);
        }
    }
    g<<rez<<"\n";
    g<<k<<"\n";
    for(i=1;i<=k;i++)
      g<<a[fin[i]].x<<" "<<a[fin[i]].y<<"\n";
    return 0;
}