Cod sursa(job #1426263)

Utilizator R.C-VRadu Cristian Valentin R.C-V Data 29 aprilie 2015 11:50:53
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.23 kb
#include<stdio.h>
#include<algorithm>
#include<vector>

using namespace std;

const int maxim = 400100;
int *C;

int grupa(int i,int G[])
{
    if (G[i] == i) return i;
    G[i] = grupa(G[i],G);
    return G[i];
}

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

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

int main()
{
    int *G, *X, *Y, *indice;
    G=new int[maxim];
    X=new int[maxim];
    Y=new int[maxim];
    C=new int[maxim];
    indice=new int[maxim];
    int N,M,CT=0;;
    vector<int> V;
    freopen("apm.in","r",stdin);
    freopen("apm.out","w",stdout);
    scanf("%d %d\n",&N,&M);
    for(int i = 1;i <= M;++i)
    {
        scanf("%d %d %d\n",&X[i],&Y[i],&C[i]);
        indice[i] = i;
    }

    for(int i = 1;i <= N; ++i) G[i] = i;
    sort(indice + 1,indice + M + 1,comp);
    for(int i = 1;i <= M; ++i)
    {
        if (grupa(X[indice[i]],G) != grupa(Y[indice[i]],G)){
            CT += C[indice[i]];
            reuniune(X[indice[i]],Y[indice[i]],G);
            V.push_back(indice[i]);
        }
    }
    printf("%d\n",CT);
    printf("%d\n",N - 1);
    for(int i = 0;i < N - 1; ++i)
        printf("%d %d\n",X[V[i]],Y[V[i]]);
    return 0;
}