Cod sursa(job #1497712)

Utilizator superstar1998Moldoveanu Vlad superstar1998 Data 7 octombrie 2015 10:37:30
Problema Arbore partial de cost minim Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.33 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]++;
    t[multime(i)]=multime(j);
}
inline bool 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=1;k<=m;k++)
    {
        i=M[k].x;
        j=M[k].y;
        c=M[k].c;
        if(multime(i)!=multime(j))
        {
            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;
}