Cod sursa(job #2176741)

Utilizator unknownpersonBidasca Carina Georgiana unknownperson Data 17 martie 2018 23:31:44
Problema Arbore partial de cost minim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1 kb
#include<bits/stdc++.h>
#define pb push_back

using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
const int maxn = 400100;

int GR[maxn],X[maxn],Y[maxn],C[maxn];
int N,M,ANS,IND[maxn];
vector<int> VANS;

int comp(int i)
{
    if (GR[i] == i) return i;
    GR[i] = comp(GR[i]);
    return GR[i];
}

void reuniune(int i,int j)
{
    GR[comp(i)] = comp(j);
}

bool cmpf(int i,int j)
{
    return(C[i] < C[j]);
}

int main()
{
    f>>N>>M;
    for(int i = 1;i <= M;++i)
    {
        f>>X[i]>>Y[i]>>C[i];
        IND[i] = i;
    }

    for(int i = 1;i <= N; ++i) GR[i] = i;
    for(int i = 1;i <= N; ++i) GR[i] = i;
    sort(IND + 1,IND + M + 1,cmpf);
    for(int i = 1;i <= M; ++i)
    {
        if (comp(X[IND[i]]) != comp(Y[IND[i]])){
            ANS += C[IND[i]];reuniune(X[IND[i]],Y[IND[i]]);
            VANS.pb(IND[i]);
        }
    }
    g<<ANS<<"\n";
    for(int i = 0;i < N - 1; ++i)
        g<<X[VANS[i]]<<" "<<Y[VANS[i]]<<"\n";
    return 0;
}