力扣:743.网络延迟时间(dijkstra最短路径算法)
1、 问题描述
有 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) 对都 互不相同(即,不含重复边)
链接:https://leetcode-cn.com/problems/network-delay-time
2、dijkstra
dijkstra求最短路径的基本思路为,从起点开始,逐步更新起点到每个点的最短距离,直到求出目标点。
可以理解为一个带dp table 的bfs算法。不断更新distTo数组。
3、题目解答
class Solution {
public:
int networkDelayTime(vector<vector<int>>& times, int n, int k) {
vector<vector<int>> graph(n+1,vector<int>(n+1,-1));
for(int i = 0; i< times.size(); i++){//构造邻接矩阵
int from = times[i][0];
int to = times[i][1];
int cost = times[i][2];
graph[from][to] = cost;
}
vector<int> distTo(n+1, INT_MAX);// 起点到每个点的最短距离
queue<int> q;
q.push(k);
distTo[k] = 0;
while(!q.empty()){
int cur = q.front();q.pop();
for(int i = 1; i <= n; i++){
if(graph[cur][i] != -1){//如果可达
int to = i;
int cost = graph[cur][i];
if(distTo[i] > distTo[cur] + cost){//更新最短距离
distTo[i] = distTo[cur] + cost;
q.push(i);
}
}
}
}
int res = 0;
for(int i = 1; i <= n; i++){
if(distTo[i] == INT_MAX)return -1;
res = max(res, distTo[i]);
}
return res;
}
};
res = max(res, distTo[i]);
}
return res;
}
};