机试真题 弗洛伊德算法

#include<iostream>
#include<vector>
#include<string>
using namespace std;

const int maxn = 210;
const int INF = 10000000;
int ma[maxn][maxn];

int n, m;
int st, ed;

void init() {
	fill(ma[0], ma[0] + maxn * maxn, INF);
	for (int i = 0; i < n; i++) {
		ma[i][i] = 0;
	}
}

void floyd() {
	for (int k = 0; k < n; k++) {
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				if (ma[i][k] != INF && ma[k][j] != INF && ma[i][k] + ma[k][j] < ma[i][j]) {
					ma[i][j] = ma[i][k] + ma[k][j];
				}
			}
		}
	}
}

int main() {
	while (cin >> n >> m) {
		init();
		int a, b, c;
		for (int i = 0; i < m; i++) {
			cin >> a >> b >> c;
			ma[a][b] = ma[b][a] = c;
		}
		cin >> st >> ed;
		floyd();
		if (ma[st][ed] == INF)
			cout << -1 << endl;
		else
			cout << ma[st][ed] << endl;
	}
}