AcWing 849. Dijkstra求最短路 I
m的数量级在n^2的级别,所以该最短路问题使用朴素版的Dijkstra算法。
代码:
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 510;
int n, m;
int g[N][N];//该图为稠密图,所以使用邻接矩阵来存
int dist[N];//每个点距起点的距离
bool st[N];//记录当前点是否被处理过(已经距离起点是最近的了)
int dijsktra() {
memset(dist, 0x3f, sizeof dist);//将每个点距离初始化为正无穷
dist[1] = 0;//1号点距离初始化为0
//迭代n步
for (int i = 0; i < n; ++i) {
//寻找未被处理过的,距离最近的点
int t = -1;
for (int j = 1; j <= n; ++j)
if (!st[j] && (t == -1 || dist[t] > dist[j]))
t = j;
st[t] = true;//将t加入已经被处理过的集合
//使用t更新当前每个点距起点的距离
for (int j = 1; j <= n; ++j)
dist[j] = min(dist[j], dist[t] + g[t][j]);//使用 起点到t的距离加t到j的距离 更新 起点到j的距离
}
if (dist[n] == 0x3f3f3f3f) return -1;//如果n这点的距离为正无穷,那就说明不存在从1到n的路径
return dist[n];
}
int main() {
scanf("%d%d", &n, &m);
memset(g, 0x3f, sizeof g);
while (m--) {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
g[a][b] = min(g[a][b], c);//维护最小的,因为该题可能有重边
}
int t = dijsktra();
printf("%d\n", t);
return 0;
}