Cod sursa(job #1216376)

Utilizator EpictetStamatin Cristian Epictet Data 4 august 2014 13:39:35
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.32 kb
// Algoritmul lui Kruskal
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
int n, m, minim, maxim, sol, C[200010];
struct Muchie { int x, y, cost; } V[400010];
vector < int > S;

inline void Citire()
{
    fin >> n >> m;
    for (int i=1; i<=m; i++)
        fin >> V[i].x >> V[i].y >> V[i].cost;
    for (int i=1; i<=n; i++) C[i] = i;
}

inline void SortareMuchii(int st, int dr)
{
    if (st < dr)
    {
        Muchie x = V[st];
        int i = st;
        int j = dr;
        while (i < j)
        {
            while (i < j && V[j].cost >= x.cost) j--;
            V[i] = V[j];
            while (i < j && V[i].cost <= x.cost) i++;
            V[j] = V[i];
        }
        V[i] = x;
        SortareMuchii(st, i-1);
        SortareMuchii(i+1, dr);
    }
}

inline int Contr(int i)
{
    if (C[i] == i) return i;
    C[i] = Contr(C[i]);
    return C[i];
}

int main()
{
    Citire();
    SortareMuchii(1, m);

    for (int i=1; i<=m; i++)
    {
        if (Contr(V[i].x) != Contr(V[i].y))
        {
            S.push_back(i);
            sol += V[i].cost;
            C[Contr(V[i].x)] = Contr(V[i].y);
        }
    }

    fout << sol << '\n' << S.size() << '\n';
    for (int i=0; i<S.size(); i++)
        fout << V[S[i]].x << ' ' << V[S[i]].y << '\n';

    fout.close();
    return 0;
}