【数据结构】Dijkstra算法 | 迪杰斯特拉 | 最短路问题

Dijkstra

Dijkstra 算法是求一个图中一个点到其他所有点的最短路径的算法。

基本步骤

一、初始化数据

1. 使用邻接表/邻接矩阵存储路径关系（起名为mp）
2. 初始化起点到所有点的距离（起名为dis）—-与自己距离设为0，与其它点设为无限大（0x7ffff）
3. 设置vis数组并初始化为0，用于记录访问过的点

二、Dijkstra

1. 遍历当前点（第一次为起点）的所有相邻点（mp数组中和当前点有关联的点）
2. 找出其中最小的那个，进行dis[v] = dis[v] < dis[u] + mp[u][v] ? dis[v] : dis[u] + mp[u][v];
3. 将满足更新条件的点依次排列在后，作为预备的当前点。
4. 如果找过的点就不需要再找了（用vis标记）
5. 循环1-4直到遍历完所有点（如果有目标，可以找到目标点跳出）

dijkstra+堆优化模板

堆优化就是通过优先队列的堆排序，避免了遍历找最小值的时间。

代码

#include<bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
int N, K, M, S, T, u, v, d;
int mp[110][110];
bool vis[110];
int dis[110];

struct Node {
		int num, Dist;
		// 运算符重载使优先队列按照dis递增顺序排列
		bool operator < (const Node& x)const {
			return x.Dist < Dist;
		}
};

void Dijkstra(int s) {
		memset(vis, 0, sizeof(vis));
		memset(dis, INF, sizeof(dis));
		dis[s] = 0;
		priority_queue<Node>q;
		q.push({ s, dis[s] });
		while (!q.empty()) {
				int u = q.top().num;
				q.pop();
				if (vis[u])continue;
				vis[u] = true;
				for (int v = 1; v <= N; v++) {
						if (!vis[v] && mp[u][v]) {
								if (dis[u] + mp[u][v] < dis[v]) {
										dis[v] = dis[u] + mp[u][v];
										q.push({ v, dis[v] });
								}
						}
				}
		}
}

int main() {
		// 邻接矩阵建边
		for (int i = 1; i <= M; i++) {
				cin >> u >> v >> d;
				mp[u][v] = mp[v][u] = d;
		}
		Dijkstra(S);
		if (dis[T] >= 0x3f3f3f3f) puts("-1");
		else printf("%d", dis[T]);
}