Cod sursa(job #2454424)

Utilizator mvcl3Marian Iacob mvcl3 Data 8 septembrie 2019 14:19:27
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include<fstream>
#include<vector>
#include<string>
#include<algorithm>

const std::string inFileName =	"royfloyd.in";
const std::string outFileName = "royfloyd.out";
const int MAX_WEIGHT_VALUE = 1001;

inline void readData(std::ifstream& in, int& n, std::vector<std::vector<int>>& rf) {
	for (int i = 0; i < n; ++i) {
		for (int j = 0; j < n; ++j) {
			in >> rf[i][j];
		}
	}
}

inline void solve(const int& n, std::vector<std::vector<int>>& rf) {
	for (int k = 0; k < n; ++k) {
		for (int i = 0; i < n; ++i) {
			for (int j = 0; j < n; ++j) {
				if (i != j && rf[i][k] && rf[k][j] && (rf[i][j] > rf[i][k] + rf[k][j] || !rf[i][j])) {
					rf[i][j] = rf[i][k] + rf[k][j];
				}
			}
		}
	}
}

int main()
{
	std::ifstream in(inFileName);
	std::ofstream out(outFileName);
	
	int n;
	in >> n;
	std::vector<std::vector<int>> rf(n, std::vector<int>(n));
	readData(in, n, rf);
	solve(n, rf);

	for (int i = 0; i < n; ++i) {
		for (int j = 0; j < n; ++j) {
			out << rf[i][j] << ' ';
		}
		out << std::endl;
	}

	return 0;
}