Cod sursa(job #2935962)

Utilizator stefan24Sandor Stefan stefan24 Data 7 noiembrie 2022 19:01:51
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.12 kb
#include <bits/stdc++.h>
using namespace std;
#define nmax 200000
#define mmax 400000
ifstream f("apm.in");
ofstream g("apm.out");

int t[nmax+5],rang[nmax+5],a[mmax+5];

struct graf{
    int x,y,cost;
}v[mmax+5];

bool comp(graf a,graf b){
    return(a.cost<b.cost);
}

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

void unire(int a,int b){
    int ra=radacina(a);
    int rb=radacina(b);
    if(ra!=rb){
        if(rang[ra]>rang[rb])t[rb]=ra;
        else{
            t[ra]=rb;
            if(rang[ra]==rang[rb])rang[rb]++;
        }
    }
}

int main()
{
    int n,m,m1=0;
    f>>n>>m;
    for(int i=1;i<=m;++i)
        f>>v[i].x>>v[i].y>>v[i].cost;
    sort(v+1,v+m+1,comp);
    int s=0;
    for(int i=1;i<=m;++i)
    {
        int rx=radacina(v[i].x);
        int ry=radacina(v[i].y);
        if(rx!=ry){
            s+=v[i].cost;
            a[++m1]=i;
            unire(rx,ry);
        }
    }
    g<<s<<"\n"<<m1<<"\n";
    for(int i=1;i<=m1;++i)
        g<<v[a[i]].x<<" "<<v[a[i]].y<<"\n";
}