思想:贪心。将所有节点分成两类:已确定从起点到当前点的最短路径长度的节点,以及未确定从起点到当前点的最短路长度的节点。(下面简称「未确定节点」和「已确定节点」)。
每次从「未确定节点」中取一个与起点距离最短的点,将它归类为「已确定节点」,并用它「更新」从起点到其他所有「未确定节点」的距离。直到所有点都被归类为「已确定节点」。
数据:
(1)集合S:已求得的最短路径顶点,初始化把源点v0放入
(2)数组dist[]:从源点v0到顶点i当前的最短路径长度,初始化为MAX
(3)arcs[0][i]:图的邻接矩阵
(4)集合V:图的所有顶点
步骤:
(1)初始化:集合S为{v0},dist[i] = arcs[0][i]
(2)从V-S中选出最短dist[j],将j加入S
(3)更新v0到V-S中任一顶点k的最短路径,dist[k] = min(dist[k], dist[j] + arcs[j][k])
例题:
https://leetcode-cn.com/problems/network-delay-time/
743. 网络延迟时间
有 n 个网络节点,标记为 1 到 n。
给你一个列表 times,表示信号经过 有向 边的传递时间。 times[i] = (ui, vi, wi),其中 ui 是源节点,vi 是目标节点, wi 是一个信号从源节点传递到目标节点的时间。
现在,从某个节点 K 发出一个信号。需要多久才能使所有节点都收到信号?如果不能使所有节点收到信号,返回 -1 。
示例 1
输入:times = [[2,1,1],[2,3,1],[3,4,1]], n = 4, k = 2
输出:2
示例 2
输入:times = [[1,2,1]], n = 2, k = 1
输出:1
示例 3
输入:times = [[1,2,1]], n = 2, k = 2
输出:-1
提示:
1 <= k <= n <= 100
1 <= times.length <= 6000
times[i].length == 3
1 <= ui, vi <= n
ui != vi
0 <= wi <= 100
所有 (ui, vi) 对都 互不相同(即,不含重复边)
class Solution {
public:
int networkDelayTime(vector<vector<int>>& times, int n, int k) {
const int inf = 101;
//邻接矩阵
vector<vector<int>> g(n, vector<int>(n, inf));
for (auto &t : times) {
int x = t[0] - 1, y = t[1] - 1;//结点标记为1到n
g[x][y] = t[2];
}
//k到结点i的最短距离,初始化为最大值
vector<int> dist(n, inf);
dist[k - 1] = 0;
vector<int> used(n);//已求得的最短路径长度
for (int i = 0; i < n; i++) {
int x = -1;
//未纳入的当前最短路径
for (int y = 0; y < n; y++) {
if (!used[y] && (x == -1 || dist[y] < dist[x])) x = y;
}
used[x] = true;
//更新
for (int y = 0; y < n; y++) {
dist[y] = min(dist[y], dist[x] + g[x][y]);
}
}
int ans = *max_element(dist.begin(), dist.end());
return ans == inf ? -1 : ans;
}
};