Cod sursa(job #2545250)

Utilizator TeddyDinutaDinuta Eduard Stefan TeddyDinuta Data 12 februarie 2020 22:28:19
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.25 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;
}