一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第17天,点击查看活动详情。
算法步骤
- 初始化距离
- dist[1] = 0 :1号点到起点的距离是0
- dist[i] = +∞:其他所有点到起点的距离都是正无穷
- 因为第一步只有起点被遍历到了,其他点都还没有被遍历到
- 迭代循环,循环n次
- 集合S:当前已经确定最短距离的点放到集合S里面去
- 第一步:找到不在S中的距离最近的点,赋值给t
- 第二步:将t加到S里面去
- 第三步:用t更新其他所有点的距离
由于这个边数较多,稠密图
所以要用邻接矩阵来存
题目
代码
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 510;
int n, m; // 点数和边数
int g[N][N]; // 稠密图,用临界矩阵来存
int dist[N]; // dist:表示从1号点走到其他点的距离
bool st[N]; // 表示每个点的最短路是否已经确定了
int dijkstra()
{
// 初始化一下所有距离
memset(dist, 0x3f, sizeof dist);
dist[1] = 0;
// 迭代n次
for (int i = 0; i < n; i++)
{
int t = -1;
for (int j = 1; j <= n; j++) // 这段代码的作用:在st == false的所有点中找到dist点最小的点即可
if (!st[j] && (t == -1 || dist[t] > dist[j])) // 当前点没有确定最短路,t == -1 || 当前这个t不是最短的
t = j; // 将j赋值给t
st[t] = true; // 将t加到集合里面取
// 用t更新一下所有点的距离
for (int j = 1; j <= n; j++)
dist[j] = min(dist[j], dist[t] + g[t][j]);
}
if (dist[n] == 0x3f3f3f3f) return -1; // 说明不存在最短距离
return dist[n]; // 否则返回dist[n]
}
int main()
{
scanf("%d%d", &n, &m); // 读入点数和边数
// 初始化邻接矩阵
/* for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
if (i == j) g[i][j] = 0;
else g[i][j] = INF;*/
memset(g, 0x3f, sizeof g);
// 读入m条边
while (m--)
{
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
g[a][b] = min(g[a][b], c); //因为可能存在重边,取最短的那条边
}
int t = dijkstra();
printf("%d\n", t);
return 0;
}