Cod sursa(job #2954274)

Utilizator TeddyDinutaDinuta Eduard Stefan TeddyDinuta Data 13 decembrie 2022 19:07:29
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.35 kb
#include <bits/stdc++.h>

using namespace std;
ifstream in("apm.in");
ofstream out("apm.out");
int n, m, p[200100], h[200100], cmin, k;
vector<pair<int, int>> ans;

struct muchie
{
    int x, y, c;
    bool operator<(const muchie &aux) const
    {
        return c < aux.c;
    }
}v[400005];

int find(int x)
{
    int r = x;
    while (r != p[r]) r = p[r];
    while (p[x] != r)
    {
        int aux = p[x];
        p[x] = r ;
        x = p[x];
    }
    return r;
}

void Union(int x, int y)
{
    int rx = find(x);
    int ry = find(y);
    if (rx != ry)
    {
        if (h[rx] < h[ry]) p[rx] = ry;
        else if (h[rx] > h[ry]) p[ry] = rx;
        else
        {
            p[rx] = ry;
            h[ry] += h[rx];
        }
    }
}

void apm()
{
    for(int i = 1;i <= n;i++) {
        p[i] = i;
        h[i] = 1;
    }

    for(int i = 1; i <= m && k < n-1; i++)
        if(find(v[i].x) != find(v[i].y))
       {
          Union(v[i].x, v[i].y);
          cmin += v[i].c;
          k++;
          ans.push_back({v[i].x, v[i].y});
       }
}

int main()
{
    in>>n>>m;
    for(int i = 1; i <= m; i++)
        in>>v[i].x>>v[i].y>>v[i].c;
    sort(v + 1, v + m + 1);
    apm();
    out<<cmin<<'\n';
    out<<k<<'\n';
    for(auto it : ans)
        out<<it.first<<" "<<it.second<<'\n';
    return 0;
}