Floyd (多源最短路径)

Floyd (多源最短路径)

const int INF = 0x3f3f3f3f; // 无穷大 
vector<vector<int> > dist; // 距离向量 
vector<vector<int> > graph; // 邻接矩阵 
int n; // 结点个数 
int m; // 边数 
int u; // 开始点 
int v; // 结束点 
int w; // 权值 

// 插点法
void floyd(vector<vector<int> > graph) {
	// 初始距离向量 
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			dist[i][j] = graph[i][j];
		}
	}
	// 计算多源最短路径 
	for (int k = 0; k < n; k++) {
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				// 插点法
				// 间接到达距离 < 直接到达距离 
				if (dist[i][k] + dist[k][j] < dist[i][j]) {
					dist[i][j] = dist[i][k] + dist[k][j];
				}
			}
		}
	}	
} 
// 预处理 
void init() {
	ifstream infile("in.txt", ios::in);
	if (!infile.is_open()) {
		cout << "file opened fail" << endl;
	}
	infile >> n >> m;
	
	graph.resize(n, vector<int> (n, INF));
	
	dist.resize(n, vector<int> (n, INF));
	
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			if (i == j)
				graph[i][j] = 0;
		}
	}
	for (int j = 0; j < m; j++) {
		infile >> u >> v >> w;
		graph[u][v] = w;
	}
	infile.close();
}
// 打印
void print() {
	ofstream outfile("out.txt", ios::out);
	for (int i = 0; i < n ;i++) {
		for (int j = 0; j < n; j++) {
			outfile << dist[i][j] << " ";
		}
		outfile << endl;
	}
	outfile.close(); 
} 
int main() {
	// 预处理 
	init();
	// 执行算法 
	floyd(graph);
	// 输出到文件 
	print();

	return 0;
}
/*
案例数据 
4 8
0 1 1
0 3 4
1 2 9
1 3 2
2 3 8
3 2 6
2 1 5
2 0 3
*/