单源最短路径Dijkstra

最短路径

解题思路

Dijkstra (单源最短路径)

const int INF = INT_MAX; // 无穷大 
int n; // 顶点个数 
int m; // 边数 
int u; // 开始点 
int v; // 结束点 
int w; // 权值 
vector<vector<int> > graph; // 邻接矩阵 
vector<int> dist; // 距离向量 
vector<bool> vis; // V - S集合 
vector<int> path; // 路径向量 
// dijkstra算法 
void Dijkstra(vector<vector<int> > graph, int u) {
	// 距离向量预处理 
	for (int i = 0; i < n; i++) {
		dist[i] = graph[u][i];
		vis[i] = false;
		// 路径向量初始化 
		if (dist[i] == INF) {
			path[i] = -1;
		} else {
			path[i] = u;
		}
	}
	// 源点到自己的距离为 0  
	dist[u] = 0;
	// 进入 V 集合， 表示已选取 
	vis[u] = true;
	// 一次选取 n 个顶点 
	for (int i = 0; i < n; i++) {
		int temp = INF;
		int t = u;
		// 选取距离源点距离最短的顶点 
		for (int j = 0; j < n; j++) {
			if (!vis[j] && dist[j] < temp) {
				temp = dist[j];
				t = j;
			}
		}
		// 没有找到 直接返回 
		if (u == t) return;
		// 进入 V 集合 
		vis[t] = true;
		// 间接到达 < 直接到达 更新距离向量 
		for (int j = 0; j < n; j++) {
			if (graph[t][j] < INF && !vis[j] && (dist[t] + graph[t][j] < dist[j])) {
				dist[j] = dist[t] + graph[t][j];
				// 记录路径 
				path[j] = t;
			}
		}
	}
} 
int main() {
	// 读取文件 
	ifstream infile("in.txt", ios::in);
	// 写入文件 
	ofstream outfile("out.txt", ios::out);
	
	infile >> n >> m;
	 
	graph.resize(n, vector<int> (n, INF));
	dist.resize(n);
	vis.resize(n);
	path.resize(n);
	// 建立邻接矩阵 
	for (int i = 0; i < m; i++) {
		infile >> u >> v >> w;
		graph[u][v] = w;
	} 
	// 执行算法 
	Dijkstra(graph, 0);
	// 输出数据到文件中 
	outfile << "-----源点 0 到各点的最短路径长度-----" << endl; 
	for (int i = 0; i < n; i++) {
		outfile << "源点 0 到 " << i << "点的最短距离为：" << dist[i] << endl;
	}
	outfile << endl << "--------最短路径路线--------" << endl; 
	for (int i = 0; i < n; i++) {
		int k = i;
		string way = "";
		while (path[k] != -1) {
			way = "-->" + to_string(k) + way; 
			k = path[k];
		}
		way = "0" + way;
		outfile << "源点 0 到 " << i << "点的最短路径路线为："  << way << endl;
	}
}
/*
测试数据 
5 8
0 1 2
0 2 5
1 2 2
1 3 6
2 3 7
3 2 2
2 4 1
3 4 4
*/