Cod sursa(job #2656609)

Utilizator Snake2003lalallalal Snake2003 Data 8 octombrie 2020 09:01:19
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.59 kb
#include <iostream>
#include <fstream>
#include <algorithm>

#define Nmax 200005
#define Mmax 400005

using namespace std;

ifstream fin("apm.in");
ofstream fout("apm.out");

int Tata[Nmax], Rank[Mmax];

pair < int, int > Perechi[Mmax];

struct elemente
{
    int x, y, cost;

}V[Mmax];

bool Crescator(elemente a, elemente b)
{
    return a.cost < b.cost;
}

int Gasit( int Nod )
{
    while( Nod != Tata[Nod] )
        Nod = Tata[Nod];

    return Nod;
}

void Unire( int a, int b )
{
    if( Rank[a] < Rank[b] )
        Tata[a] = b;
    else if( Rank[a] > Rank[b] )
        Tata[b] = a;
    else if( Rank[a] == Rank[b] )
    {
        Tata[a] = b;
        Rank[b] ++;
    }
}

int main()
{
    long long Suma = 0;
    int k = 0;
    int vf, muchii;
    fin >> vf >> muchii;
    for(int i = 1; i <= muchii; i ++)
    {
        fin >> V[i].x >> V[i].y >> V[i].cost;
    }
    for(int i = 1; i <= vf; i ++)
    {
        Tata[i] = i;
        Rank[i] = 1;
    }

    sort( V + 1, V + muchii + 1, Crescator );

    for(int i = 1; i <= muchii; i ++)
    {
        int tata_x = Gasit(V[i].x);
        int tata_y = Gasit(V[i].y);

        if( Tata[tata_x] != Tata[tata_y] )
        {
            Unire( tata_x, tata_y );

            Suma += V[i].cost;
            Perechi[++ k].first = V[i].x;
            Perechi[k].second= V[i].y;
        }
    }

    fout << Suma << "\n";

    fout << vf - 1 << "\n";

    for(int i = 1; i < vf; i ++)
    {
        fout << Perechi[i].second << " " << Perechi[i].first << "\n";
    }

    return 0;
}