Cod sursa(job #3165996)

Utilizator rarestudurTudur Rares rarestudur Data 7 noiembrie 2023 12:59:14
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.44 kb
#include <algorithm>
#include <fstream>
#include <vector>
#include <bitset>
using namespace std;

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

const int MMAX = 4e5 + 1;
const int N = 2e5 + 1;
int t[N],nr[N],n,m;
bitset <MMAX> folosit;


struct muchie
{
    int x,y,c;
};

muchie M[MMAX];


bool cmp(muchie m1, muchie m2)
{
    return(m1.c < m2.c);
}

int radacina(int x)
{
    if(t[x] == 0)
    {
        return x;
    }
    t[x] = radacina(t[x]);
    return t[x];
}

bool verificare(int x,int y)
{
    return (radacina(x) == radacina(y));
}

void reuniune(int x,int y)
{
    int rx = radacina(x);
    int ry = radacina(x);
    if(nr[rx] < nr[ry])
    {
        t[rx] = ry;
        nr[ry] += nr[rx];
    }
    else
    {
        t[ry] = rx;
        nr[rx] += nr[ry];
    }
}


int kruskal()
{
    sort(M,M+m,cmp);
    int cost = 0,nrm = 0,i = 0;
    while(nrm < n-1)
    {
        if(!verificare(M[i].x,M[i].y))
        {
            folosit[i] = true;
            nrm++;
            cost += M[i].c;
            reuniune(M[i].x,M[i].y);
        }
        i++;
    }
    return cost;
}


int main(int argc, const char * argv[])
{
    in >> n >> m;
    for(int i = 0; i < m; i++)
    {
        in >> M[i].x >> M[i].y >> M[i].c;
    }
    out << kruskal() << "\n" << n - 1 << "\n";
    for(int i = 0; i < m; i++)
    {
        if(folosit[i])
        {
            out << M[i].y << " " << M[i].x << "\n";
        }
    }
    in.close();
    out.close();
    return 0;
}