Cod sursa(job #2281010)

Utilizator filicriFilip Crisan filicri Data 11 noiembrie 2018 15:19:02
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.99 kb
#include <fstream>
#include <algorithm>
#define MMAX 400004
#define NMAX 200004
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
int n,m,cc[NMAX],sol[NMAX],ct;
int find (int x)
{
    if (cc[x]==x) return x;
    return cc[x]=find(cc[x]);
}
void reuniune (int x,int y)
{
    cc[find(x)]=find(y);
}
struct s
{
    int x,y,cost;
    bool operator < (const s &other) const
    {
        return cost < other.cost;
    }
} v[NMAX];
int main()
{
    f>>n>>m;
    for (int i=1;i<=m;i++)
    {
        f>>v[i].x>>v[i].y>>v[i].cost;
    }
    for (int i=1;i<=n;i++) cc[i]=i;
    sort(v+1,v+m+1);
    for (int i=1;i<=m;i++)
    {
        if (find(v[i].x)!=find(v[i].y))
        {
            reuniune(v[i].x,v[i].y);
            ct+=v[i].cost;
            sol[0]++;
            sol[sol[0]]=i;
        }
    }
    g<<ct<<'\n'<<sol[0]<<'\n';
    for (int i=1;i<=sol[0];i++) g<<v[sol[i]].x<<' '<<v[sol[i]].y<<'\n';
    f.close();
    g.close();
    return 0;
}