Cod sursa(job #2756435)

Utilizator filipinezulBrain Crash filipinezul Data 31 mai 2021 18:34:03
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.44 kb
#include <fstream>
#include <algorithm>
#include <bitset>

using namespace std;

const int N=200001;
const int M=400001;

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

int t[N], nr[N], n, m;
muchie v[M];
bitset <N> folosit;

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(y);
    if (nr[rx] <nr[ry])
    {
        t[rx] = ry;
        nr[ry] += nr[rx];
    }
    else
    {
        t[ry]= rx;
        nr[rx] += nr[ry];
    }
}

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

int main()
{
    ifstream in("apm.in");
    ofstream out("apm.out");
    in>>n>>m;
    for(int i=0; i<m; i++)
    {
        in>>v[i].x >> v[i].y >> v[i].c;
    }
    in.close();
    out<<kruskal()<<"\n"<<n-1<<"\n";
    for(int i=0; i<m; i++)
    {
        if(folosit[i])
        {
            out<<v[i].x<<" "<<v[i].y<<"\n";
        }
    }
    return 0;
}