[算法] 最短路径-Floyd

Floyd算法–多源最短路径算法

可计算一个图中任意结点间的最短路径，有向图和无向图都适用。

其思想：对于计算从结点i到结点j的最短路径，需查找图G中所有可以中转的结点k，若中转后的路径e[i][k]+e[k][j]小于原始路径e[i][j]，则e[i][j]更新为e[i][k]+e[k][j]。这样把所有的可作为中转点的k都判断一遍，各结点间的最短路径长度即可求得。

如对于如下无向图：

原始边表：

第一轮以结点1为中转点：
e[2][3]更新为e[2][1]+e[1][3] = 5+2 = 7
后续同理。

第二轮以结点2为中转点：

第三轮以结点3为中转点：

第四轮以结点4为中转点：

第五轮以结点5为中转点：

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

const int inf = 999999999;

int main () {
	int n, m,  t1, t2, t3;
	cin >> n >> m;			// n:结点总数1-n，m:边总数 
	vector<vector<int>> e;	// 邻接矩阵 边表 
	e.resize(n+1, vector<int>(n+1));
	// 初始化边表 
	for (int i = 1; i < n+1; i++) {
		for (int j = 1; j < n+1; j++) {
			if (i == j) {
				e[i][j] = 0;
			} else {
				e[i][j] = inf;
			}
		}
	}
	for (int i = 0; i < m; i++) {
		cin >> t1 >> t2 >> t3;
		// 无向图 
		e[t1][t2] = t3;
		e[t2][t1] = t3;
	}
	// Floyd算法
	for (int i = 1; i < n+1; i++) {
		for (int j = 1; j < n+1; j++) {
			for (int k = 1; k < n+1; k++) {
				if (e[i][k] < inf && e[k][j] < inf && e[i][j] > e[i][k] + e[k][j]) {
					e[i][j] = e[i][k] + e[k][j];
				}
			}
		}
	} 
	cout << endl;
	// 输出各结点间的最短路径长度
	for (int i = 1; i < n+1; i++) {
		for (int j = i+1; j < n+1; j++) {
			cout << i << " " << j << " " << e[i][j] << endl;
		}
	}
	return 0;
}

时间复杂度O(N^3)

input:

5 6
1 2 5
1 3 2
1 4 6
2 4 12
3 5 8
4 5 3

output:

1 2 5
1 3 2
1 4 6
1 5 9
2 3 7
2 4 11
2 5 14
3 4 8
3 5 8
4 5 3