Cod sursa(job #2346846)

Utilizator TeoMiliMilitaru Teodora TeoMili Data 18 februarie 2019 10:19:24
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.05 kb
#include <fstream>
#include <algorithm>
using namespace std;
int tata[400001],n,m,i,j,k,S,r1,r2;
struct numar{
    int x,y,c,viz;
}v[400001];
int Tata(int x){
    while(tata[x]>0)
        x=tata[x];
    return x;
}
int cmp(numar a,numar b){
    return(a.c<b.c);
}

int main()
{
    ifstream cin("apm.in");
    ofstream cout("apm.out");
    cin>>n>>m;
    for(i=1;i<=m;i++)
        cin>>v[i].x>>v[i].y>>v[i].c;
    for(i=1;i<=n;i++)
        tata[i]=-1;
    sort(v+1,v+m+1,cmp);
    k=0;
    S=0;
    for(i=1;i<=m && k<n;i++){
        r1=Tata(v[i].x);
        r2=Tata(v[i].y);
        if(r1!=r2){
            v[i].viz=1;
            S=S+v[i].c;
            k++;
            if(tata[r1]<tata[r2]){
                tata[r1]+=tata[r2];
                tata[r2]=r1;
            }
            else{
                tata[r2]+=tata[r1];
                tata[r1]=r2;
            }
        }
    }
    cout<<S<<'\n'<<k<<'\n';
    for(i=1;i<=m;i++)
        if(v[i].viz==1)
            cout<<v[i].x<<" "<<v[i].y<<'\n';
    return 0;
}