Cod sursa(job #2887173)

Utilizator iuliavIulia Vincze iuliav Data 8 aprilie 2022 22:26:41
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.1 kb
#include<iostream>
#include<fstream>
#include<algorithm>
#include<vector>

using namespace std;

const int maxim = 4e5+1;

int n,m,s,cost_sort[maxim],comp_conexa[maxim],x[maxim],y[maxim],c[maxim];
vector<int> arbore;

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

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

bool cmp_function(int i,int j)
{
	return c[i]<c[j];
}

int main()
{
    freopen("apm.in","r",stdin);
	freopen("apm.out","w",stdout);
	scanf("%d %d\n",&n,&m);
    int i;
	for(i=1;i<=m;i++)
	{
		scanf("%d %d %d\n",&x[i],&y[i],&c[i]);
		cost_sort[i] = i;
	}
	for(i=1;i<=n;i++)
        comp_conexa[i] = i;
	sort(cost_sort+1,cost_sort+m+1,cmp_function);
	for(i=1;i<=m;i++)
	{
		if (grupa(x[cost_sort[i]])!=grupa(y[cost_sort[i]]))
		{
			s+=c[cost_sort[i]];
			reuniune(x[cost_sort[i]],y[cost_sort[i]]);
			arbore.push_back(cost_sort[i]);
		}
	}
	printf("%d\n",s);
	printf("%d\n",n-1);
	for(i=0;i<n-1;i++)
		printf("%d %d\n",x[arbore[i]],y[arbore[i]]);
	return 0;
}