[C/C++] Dijkstra算法代码

 

如上图所示，1号节点为出发节点，6号节点为目的节点，求出发节点到目的节点的最短路径。使用邻接矩阵存储各节点之间的直接距离。

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

int dijkstra(vector<vector<int>> nums)
{
	const int n = (int)nums.size(), m = (int)nums[0].size();
	auto tupleCmp = [](tuple<int, int> l, tuple<int, int> r)
	{
		return get<1>(l) > get<1>(r);
	};
	priority_queue<tuple<int, int>, vector<tuple<int, int>>, decltype(tupleCmp)> minHeap(tupleCmp);
        // dist为距离矩阵，表示出发节点到各节点的最短距离
	vector<int> dist(m, INT_MAX), visited(m);

	dist[0] = 0;
	visited[0] = 1;
	for (int j = 1; j < m; j++)  // 距离数组赋初值
	{
		if (nums[0][j] != INT_MAX)
		{
			dist[j] = nums[0][j];
			minHeap.emplace(j, dist[j]);
		}
	}
	while (!minHeap.empty())
	{
		auto tmp = minHeap.top();
		minHeap.pop();

		const int idx = get<0>(tmp);
		if (visited[idx])  // 已访问过的节点跳过
		    continue;
		visited[idx] = 1;

		int d = get<1>(tmp);
		for (int j = 1; j < m; j++)
		{
			if (j == idx)
				continue;
			const int curDist = nums[idx][j] != INT_MAX ? d + nums[idx][j] : INT_MAX;
			if (curDist < dist[j])  // 更新距离数组
			{
				dist[j] = curDist;
				minHeap.emplace(j, curDist);
			}
		}
	}

	return dist.back();  // 返回出发节点到目的节点的最短距离
}

int main()
{
    vector<vector<int>> nums(6, vector<int>(6);
    // 邻接矩阵初始化
    nums[0][0] = 0; nums[0][1] = 1; nums[0][2] = 12;
    nums[1][1] = 0; nums[1][2] = 9; nums[1][3] = 3;
    nums[2][2] = 0; nums[2][4] = 5;
    nums[3][2] = 4; nums[3][3] = 0; nums[3][4] = 13; nums[3][5] = 15;
    nums[4][4] = 0; nums[4][5] = 4;
    nums[5][5] = 0;
    cout << dijkstra(nums) << endl;
    return 1;
}

附注：

1. Dijkstra算法是深度优先搜索算法（DFS）的一种，只能用于正边权的有向/无向图。
2. 要对邻接矩阵进行空间优化可以使用 vector<vector<int, int>> graph(6) 存储路径，第i个节点到第j个节点的距离可以表示为 graph[i][m].second ， 当graph[i][m].first == j 时。