题目等价于找到从K点出发,所有最短路径的最大值,这个时间段,所有点一定都能收到信号。所以是标准的Dijkstra算法
O(N^2)的Dijkstra思路如下:
维护一个dist数组用来存储从K点到其他所有点的最短距离,还要维护一个visited数组记录点是否已经作为中间过程用过。
最外层循环是迭代次数,对于N个点,只需要迭代N-1次,因为其实最后一个点不用再迭代了。
首先,先要找到dist数组中距离最近的一个点,这个点没有作为中间点使用过,然后去跟新其他dist数组的值
完成之后,记得到了dist数组,这里有一个技巧是直接使用inf运算来避免一些边界判断,所以inf不能取得越界。下面是这个算法的最优代码:
const int INF = 0x3f3f3f3f;
class Solution {
public:
int networkDelayTime(vector<vector<int>>& times, int N, int K) {
vector<vector<int>> matrix(N+1,vector<int>(N+1,INF));
for(auto time:times){
matrix[time[0]][time[1]] = time[2];
}
vector<bool> visited(N+1);
vector<int> dist(N+1,INF);
dist[K] = 0;
for(int i=0;i<N-1;i++){
int tmp = -1;
for(int j=1;j<=N;j++){
if(!visited[j]&&(tmp==-1||dist[j]<dist[tmp])){
tmp = j;
}
}
for(int j=1;j<=N;j++){
dist[j] = min(dist[j],dist[tmp]+matrix[tmp][j]);
}
visited[tmp] = true;
}
int ans = *max_element(dist.begin()+1, dist.end());
return ans == INF ? -1: ans;
}
};