Cod sursa(job #2690479)

Utilizator tryharderulbrebenel mihnea stefan tryharderul Data 24 decembrie 2020 09:11:51
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.15 kb
#include <bits/stdc++.h>
#define pb push_back
#define NMAX 200003
#define pb push_back

using namespace std;
struct graf{
    int x,y,cost;
};

vector<int>gr(NMAX,0);
vector<graf>v(NMAX);
vector<int>ind(NMAX);
bool cmp(int i,int j){
    return (v[i].cost<v[j].cost);
}

int grupa(int x){
    if(gr[x]==x)
        return x;
    gr[x]=grupa(gr[x]);
    return gr[x];
}

void reuniune(int i,int j){
	gr[grupa(i)] = grupa(j);
}

int main()
{
    freopen("apm.in","r",stdin);
    freopen("apm.out","w",stdout);

    int n,m;
    scanf("%d%d",&n,&m);


    for(int i=1;i<=m;i++){
        int x,y,c;
        scanf("%d%d%d",&v[i].x,&v[i].y,&v[i].cost);
        ind[i]=i;
    }

    sort(ind.begin()+1,ind.begin()+m+1,cmp);





    for(int i=1;i<=n;i++)
        gr[i]=i;
    int s=0;
    vector<int>sol;
    for(int i=1;i<=m;i++){
        if(grupa(v[ind[i]].x)!=grupa(v[ind[i]].y)){
            s=s+v[ind[i]].cost;
            reuniune(v[ind[i]].x,v[ind[i]].y);
            sol.pb(ind[i]);
        }

    }

    printf("%d\n%d\n",s,n-1);
    for(auto i:sol)
        printf("%d %d\n",v[i].x,v[i].y);


    return 0;
}