Cod sursa(job #1416524)

Utilizator karlaKarla Maria karla Data 8 aprilie 2015 12:19:58
Problema Arbore partial de cost minim Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.12 kb
#include <stdio.h>
#include <vector>
#include <algorithm>
#define MAx 400100

using namespace std;

int n, m, t[200100];
FILE*f=fopen("apm.in","r"),*g=fopen("apm.out","w");

struct graf
{
    int x, y,c;
}l[MAx];

vector <pair <int, int> > sol;

bool cmp(graf a, graf b)
{
    if(a.c > b.c)
        return 0;
    return 1;
}

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

int main()
{
    int x, y, c;
    fscanf(f,"%d %d",&n,&m);
    for(int i = 1; i <= m; i++)
    {
        fscanf(f,"%d %d %d",&l[i].x,&l[i].y,&l[i].c);
    }
    for(int i = 1; i <= n; i++)
        t[i] = i;
    sort(l + 1, l + m + 1, cmp);
    int s = 0, tx, ty, nr = 0;
    for(int i = 1; i <= m; i++)
    {
        tx = tata(l[i].x);
        ty = tata(l[i].y);
        if(tx != ty)
        {
            t[tx] = ty;
            s += l[i].c;
            nr++;
            sol.push_back(make_pair(l[i].x, l[i].y));
        }
    }
    fprintf(g,"%d\n%d\n",s,nr);
    for(int i = 0; i < nr; i++)
    {
        fprintf(g,"%d %d\n",sol[i].first, sol[i].second);
    }
    return 0;
}