Cod sursa(job #1036031)

Utilizator sebinechitasebi nechita sebinechita Data 18 noiembrie 2013 22:25:13
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.09 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;

FILE *fin=fopen("apm.in", "r"), *fout=fopen("apm.out", "w");

#define MAX 200100

typedef long long int lli;

int fr[MAX],cost,i, n, v, m, j, g, b[MAX], c[MAX];

struct sebi{
    int x,y,c;
};

sebi a[2*MAX];

inline bool cmp(sebi d, sebi f)
{
    return d.c<f.c;
}

inline int gr(int l)
{
    if(fr[l]!=l)
        fr[l]=gr(fr[l]);

    return fr[l];
}

int main()
{

    fscanf(fin, "%d %d\n", &n, &m);
    for(i=1;i<=m;i++)
    {
        fscanf(fin, "%d %d %d\n", &a[i].x, &a[i].y, &a[i].c);
    }
    for(i=1;i<=n;i++)
        fr[i]=i;
    sort(a+1,a+m+1,cmp);
    g=0;
    cost=0;
    for(i=1;g<n-1;i++)
    {
        if(gr(a[i].x)!=gr(a[i].y))
        {
            b[g]=a[i].x;
            c[g]=a[i].y;
            g++;
            cost+=a[i].c;
            fr[gr(a[i].x)]=gr(a[i].y);
        }
    }
    fprintf(fout, "%d\n%d\n", cost, n-1);

    for(i=0;i<n-1;i++)
        fprintf(fout, "%d %d\n", c[i], b[i]);

    return 0;
}