Cod sursa(job #1260550)

Utilizator seby5381Marinescu Sebastian seby5381 Data 11 noiembrie 2014 13:26:29
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include <cstdio>
#include <algorithm>
using namespace std;
int i,x,y,n,m,c,k,j,cost,t[400007],aux,xx1,yy1,nr,h[400007],s,poz[400007];

struct nod
{
    int x, y, c;
} a[400007];

bool cmp ( nod a, nod b )
{
    return a.c < b.c;
}

int tata(int x)
{
    if(t[x]!=x) t[x]=tata(t[x]);
    return t[x];
}

int main ()
{
    freopen ("apm.in", "r", stdin);
    freopen ("apm.out", "w", stdout);
    scanf ("%d %d", &n, &m);
    for (i=1; i<=m; i++)
    {
        scanf("%d %d %d", &x, &y, &c);
        a[i].x = x;
        a[i].y = y;
        a[i].c = c;
    }
    sort (a+1, a+1+m, cmp);
    /*for (i=1; i<=m; i++)
    {
        printf("%d %d %d\n",a[i].x, a[i].y, a[i].c);
    }*/
    for(i=1;i<=n;i++)
    {
        h[i]=1;
        t[i]=i;
    }
    for(i=1;i<=m;i++)
    {
        xx1=tata(a[i].x);
        yy1=tata(a[i].y);
        if(xx1!=yy1)
        {
            if(h[xx1]<h[yy1]) t[xx1]=yy1;
            else t[yy1]=xx1;
            if(h[xx1]==h[yy1]) h[yy1]++;
            nr++;
            poz[nr]=i;
            s=s+a[i].c;
            if(nr==n-1) break;
        }
    }
    printf("%d\n",s);
    printf("%d\n",nr);
    for(i=1;i<=nr;i++)
    printf("%d %d\n",a[poz[i]].x,a[poz[i]].y);
    return 0;
}