1 基础知识
朴素版dijkstra算法的关键步骤:
- 初始化d[1]=0,d[2~n]=正无穷,例如0x3f3f3f3f。
- 用集合S来表示当前已被确定最小距离的结点们。
- 遍历每一个结点:找到不在S中的且距离结点1最近的结点,记为t。将结点t加入到集合S中。看看结点t可以走到哪儿,假设可以走到x,比较dist[x]和dist[t] + edge[t][x],如果前者大于后者,则用后者去更新前者。
- d[1~n],即为结点1到结点x的最短距离,x取1~n的值。
朴素版的dijkstra算法的核心是贪心,一直找最近的,最终的结果就是最近的。
2 例题讲解
题目1:求结点1到结点n的最短路。
#include <iostream>
#include <cstring>
using namespace std;
const int N = 510;
int g[N][N];
int dist[N];
bool st[N];
int n, m;
int dijkstra() {
memset(dist, 0x3f, sizeof dist);
dist[1] = 0;
for (int i = 0; i < n; ++i) {//遍历每一个结点,总共有n个结点,需要遍历n次
int t = -1; //当前,距离结点1最近的结点且不在集合S中
for (int j = 1; j <= n; ++j) {
if (!st[j] && (t == -1 || dist[t] > dist[j])) {
t = j;
}
}
st[t] = true; //把t加入到集合S中
for (int j = 1; j <= n; ++j) {//用t取更新剩余的结点
if (!st[j]) {
dist[j] = min(dist[j], dist[t] + g[t][j]);
}
}
}
if (dist[n] == 0x3f3f3f3f) return -1;
else return dist[n];
}
int main() {
cin >> n >> m;
memset(g, 0x3f, sizeof g);
int a, b, c;
while (m--) {
cin >> a >> b >> c;
g[a][b] = min(g[a][b], c);
}
cout << dijkstra() << endl;
return 0;
}