Cod sursa(job #1260548)

Utilizator binicBinica Nicolae binic Data 11 noiembrie 2014 13:25:56
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.03 kb
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,i,x,y,c,z,x1,s,yy,t,j,b[200002],mch[200002],h[200002];
struct nod
{
    int x,y;
    short c;
}a[400002];
int tata(int x)
{
    if(b[x]!=x) b[x]=tata(b[x]);
    return b[x];
}
bool cmp(nod a, nod b)
{
    if(a.c<b.c)return 1;
    return 0;
}
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",&a[i].x,&a[i].y,&a[i].c);
    for(i=1;i<=n;i++){b[i]=i;h[i]=1;}
    sort(a+1,a+m+1,cmp);
    for(i=1;i<=m;i++)
    {
        x1=tata(a[i].x);
        yy=tata(a[i].y);
        if(x1!=yy)
        {
            if(h[x1]<h[yy]) b[x1]=yy;
            else b[yy]=x1;
            if(h[x1]==h[yy]) h[yy]++;
            s=s+a[i].c;
            z++;
            mch[z]=i;
            if(z==n-1)break;
        }
    }
    printf("%d\n",s);
    printf("%d\n",z);
    for(i=1;i<=z;i++)
        printf("%d %d\n",a[mch[i]].x,a[mch[i]].y);
    return 0;
}