Cod sursa(job #1497709)

Utilizator superstar1998Moldoveanu Vlad superstar1998 Data 7 octombrie 2015 10:28:05
Problema Arbore partial de cost minim Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include <fstream>
#include <algorithm>
#define MAX_N 200001
#define MAX_M 400001
using namespace std;
struct muchie
{
    int x,y,c;
}M[MAX_M];
bool u[MAX_M];
int m,n,t[MAX_N],rang[MAX_N];
long ct;
void citeste()
{
    ifstream f("apm.in");
    f>>n>>m;
    for(int i=1;i<=m;i++)
        f>>M[i].x>>M[i].y>>M[i].c;
    f.close();
}
int multime(int k)
{
    if(t[k]!=k)
        t[k]=multime(t[k]);
    return t[k];
}
void reuneste(int i, int j)
{
    i=multime(i);
    j=multime(j);
    if(i==j) return;
    if(rang[i]<rang[j])
        t[i]=j;
    else t[j]=i;
    if(rang[i]==rang[j])
        rang[i]++;
}
inline int comp(const muchie& a, const muchie& b)
{
    return a.c<b.c;
}
void kruskal()
{
    int i,j,k,c;
    sort(M+1,M+m+1,comp);
    for(i=1;i<=n;i++)
        t[i]=i,rang[i]=0;
    for(k=0;k<m;k++)
    {
        i=M[k].x;
        j=M[k].y;
        c=M[k].c;
        if(multime(i)==multime(j)) continue;
        reuneste(i,j);
        ct+=c;
        u[k]=true;
    }
}
void scrie()
{
    ofstream g("apm.out");
    g<<ct<<endl<<n-1<<endl;
    for(int i=1;i<=m;i++)
        if(u[i])g<<M[i].x<<" "<<M[i].y<<endl;
    g.close();
}
int main()
{
    citeste();
    kruskal();
    scrie();
    return 0;
}