Cod sursa(job #1748577)

Utilizator refugiatBoni Daniel Stefan refugiat Data 26 august 2016 14:27:56
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.09 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <bitset>
using namespace std;
ifstream si("apm.in");
ofstream so("apm.out");
struct muchie
{
    int x,y;
    int cost;
};
const int MMAX=400005;
muchie v[MMAX];
int tata[200005];
bitset<MMAX>luat;

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

bool alipire(int x,int y)
{
    int tx=radacina(x),ty=radacina(y);

    if(tx!=ty)
    {
        tata[tx]=ty;
        return true;
    }

    return false;
}

bool comp(muchie a,muchie b)
{
    return a.cost<b.cost;
}
int main()
{
    int n,m;
    si>>n>>m;
    int i;
    for(i=1;i<=n;++i)
        tata[i]=i;
    for(i=0;i<m;++i)
        si>>v[i].x>>v[i].y>>v[i].cost;
    sort(v,v+m,comp);
    int cost=0;
    for(i=0;i<m;++i)
        if(alipire(v[i].x,v[i].y))
        {
            cost+=v[i].cost;
            luat[i]=true;
        }
    so<<cost<<'\n'<<n-1<<'\n';
    for(i=0;i<m;++i)
        if(luat[i])
            so<<v[i].y<<' '<<v[i].x<<'\n';
    return 0;
}