Cod sursa(job #2329979)

Utilizator Username01Name Surname Username01 Data 27 ianuarie 2019 18:47:13
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.27 kb
#include <cstdio>
#include <algorithm>

using namespace std;
FILE *f,*g;

struct name
{
    int a,b,c;
}v[200002];

bool how(name A, name B)
{
    return (A.c<B.c);
}
int tata[200002],rang[200002];
struct
{
    int aa,bb;
}sol[200002];

int multime(int nod)
{
    int rad=nod;
    while(rad!=tata[rad])
        rad=tata[rad];
    return rad;
}
void unite(int tatax, int tatay)
{
    if(rang[tatax]<rang[tatay])
        tata[tatax]=tatay;
    else
        tata[tatay]=tatax;
    if(rang[tatax]==rang[tatay])
        ++rang[tatax];
}
int main()
{
    f=fopen("apm.in","r");
    g=fopen("apm.out","w");
    int m,n,x,y,mx,my,ss=0,cost=0;
    fscanf(f,"%d %d",&n,&m);
    for(int i=1;i<=m;++i)
        fscanf(f,"%d %d %d",&v[i].a,&v[i].b,&v[i].c);
    sort(v+1,v+m+1,how);
    for(int i=1;i<=n;++i) tata[i]=i;
    for(int i=1;i<=m;++i)
    {
        x=v[i].a;
        y=v[i].b;
        mx=multime(x);
        my=multime(y);
        if(mx!=my)
        {
            unite(mx,my);
            cost+=v[i].c;
            sol[++ss].aa=x;
            sol[ss].bb=y;
        }
    }
    fprintf(g,"%d\n%d\n",cost,ss);
    for(int i=1;i<=n;++i)
        fprintf(g,"%d %d\n",sol[i].aa,sol[i].bb);

    fclose(f);
    fclose(g);
    return 0;
}