Cod sursa(job #2507565)

Utilizator bimax145Bogdan Draghici bimax145 Data 10 decembrie 2019 14:59:36
Problema Arbore partial de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.31 kb
#include<fstream>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
struct muchie															
{
	int sursa;																
	int dest;																
	int cost;																
};
vector<muchie> graf;
vector<int> parinte;
vector<pair<int, int>> sol;

inline bool SortByCost(muchie a, muchie b) { return a.cost < b.cost; }

void kruskal(int n, int m)
{
	int i, k, x, y, total_cost;
	total_cost = i = k = 0;
	while (k < n - 1) 
	{
		if (parinte[graf[i].sursa] != parinte[graf[i].dest])
		{
			k++;
			total_cost = total_cost + graf[i].cost;
			sol.push_back(make_pair(graf[i].sursa, graf[i].dest));

			x = parinte[graf[i].dest];			
			y = parinte[graf[i].sursa];

			for (int j = 1; j <= n; j++)
				if (parinte[j] == x)
					parinte[j] = y;
		}
		i++;
	}
	fout << total_cost << "\n";
}

int main()
{
	int noduri, muchii, x, y, c;
	fin >> noduri >> muchii;
	for (int i = 0; i <= noduri; i++)
		parinte.push_back(i);

	for (int i = 0; i < muchii; i++)
	{
		fin >> x >> y >> c;
		graf.push_back(muchie{ x,y,c });
	}
	sort(graf.begin(), graf.end(), SortByCost);
	
	kruskal(noduri,muchii);

	fout << sol.size() << "\n";
	for (auto it : sol)
		fout << it.first << " " << it.second << "\n";
	return 0;
}