题目分析
有 n 个网络节点,标记为 1 到 n。
给你一个列表 times
,表示信号经过 有向 边的传递时间。 times[i] = (ui, vi, wi)
,其中 ui
是源节点,vi
是目标节点, wi
是一个信号从源节点传递到目标节点的时间。
现在,从某个节点 K
发出一个信号。需要多久才能使所有节点都收到信号?如果不能使所有节点收到信号,返回 -1 。
力扣:743. 网络延迟时间
输入:times = [ [ 2 , 1 , 1 ] , [ 2 , 3 , 1 ] , [ 3 , 4 , 1 ] ] , n = 4 , k = 2
输出:2
题目分析
如果这题没有成环,可以直接进行遍历即可求出结果。
此题的考点是 单源最短路径, 求出出发点到所有其它点的最短距离,取其中最大值即为解。 这题的节点个数只有 100 节点,可以考虑使用 Floyd ,简洁易实现。
Floyd 版
class Solution {
int N = 110, M = 6010;
int[][] w = new int[N][N];
int INF = 0x3f3f3f3f;
int n, k;
public int networkDelayTime(int[][] ts, int _n, int _k) {
n = _n; k = _k;
// 初始化邻接矩阵
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
w[i][j] = w[j][i] = i == j ? 0 : INF;
}
}
// 邻接矩阵存图
for (int[] t : ts) {
int u = t[0], v = t[1], c = t[2];
w[u][v] = c;
}
// Floyd
floyd();
// 遍历答案
int ans = 0;
for (int i = 1; i <= n; i++) {
ans = Math.max(ans, w[k][i]);
}
return ans >= INF / 2 ? -1 : ans;
}
void floyd() {
for (int p = 1; p <= n; p++) {
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
w[i][j] = Math.min(w[i][j], w[i][p] + w[p][j]);
}
}
}
}
}
Dijsktra 版
class Solution {
int N = 110, M = 6010;
int[][] w = new int[N][N];
int[] dist = new int[N];
boolean[] vis = new boolean[N];
int INF = 0x3f3f3f3f;
int n, k;
public int networkDelayTime(int[][] ts, int _n, int _k) {
n = _n; k = _k;
// 初始化邻接矩阵
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
w[i][j] = w[j][i] = i == j ? 0 : INF;
}
}
// 邻接矩阵存图
for (int[] t : ts) {
int u = t[0], v = t[1], c = t[2];
w[u][v] = c;
}
// 朴素 Dijkstra
int ans = 0;
for (int end = 1; end <= n; end++) {
if (end == k) continue;
ans = Math.max(ans, dijkstra(end));
}
return ans > INF / 2 ? -1 : ans;
}
int dijkstra(int end) {
Arrays.fill(vis, false);
Arrays.fill(dist, INF);
dist[k] = 0;
for (int p = 1; p <= n; p++) {
int t = -1;
for (int i = 1; i <= n; i++) {
if (!vis[i] && (t == -1 || dist[i] < dist[t])) t = i;
}
vis[t] = true;
for (int i = 1; i <= n; i++) {
dist[i] = Math.min(dist[i], dist[t] + w[t][i]);
}
}
return dist[end];
}
}