Cod sursa(job #2039155)

Utilizator sergiudnyTritean Sergiu sergiudny Data 14 octombrie 2017 12:04:48
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.03 kb
#include <bits/stdc++.h>
#define DM 400005
#define DN 200005
#define pb push_back
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");

struct mch{
    int nx,ny,cst;
};

int p[DN],arcLeft,n,m,a,b,c,sum;
vector<mch>ans;
mch v[DM];

bool cmp(mch a,mch b){
    return (a.cst<b.cst);
}

int root(int x){
    if(x==p[x])
        return x;
    return (p[x]=root(p[x]));
}
void _merge(int x,int y){
    p[root(y)]=root(x);
}

int main()
{
    fin>>n>>m;
    arcLeft = m;
    for(int i=1;i<=m;++i){
        fin>>a>>b>>c;
        v[i]={a,b,c};
    }

    for(int i=1;i<=n;++i)
        p[i]=i;

    sort(v+1,v+m+1,cmp);
    for(int i=1;i<=m;++i){
        int a=v[i].nx,b=v[i].ny,c=v[i].cst;
        if(root(a)!=root(b)){
            _merge(a,b);
            ans.pb(v[i]);
            arcLeft--;
            sum+=c;
        }
        if(!arcLeft)
            break;
    }
    fout<<sum<<'\n'<<ans.size()<<'\n';
    for(auto i:ans)
        fout<<i.nx<<" "<<i.ny<<'\n';
    return 0;
}