题目大意:根据给出的网络图中的边集合,求出从K结点发出消息使得所有结点都接收到的传播时间。
分析:最小路径。用dijkstra算法,算出K点到每个点的最小路径长度,从中选出最大值就能保证所有节点都能够接收到消息。
代码:
class Solution {
public:
int networkDelayTime(vector<vector<int>>& times, int N, int K) {
vector<vector<int>> map(N+1,vector<int>(N+1,0x3f3f3f3f));
for(int i = 0;i < times.size();i++){
map[times[i][0]][times[i][1]] = times[i][2];
}
vector<int> dist(N+1);
vector<int> s(N+1);
for(int i = 1;i <= N;i++){
dist[i] = map[K][i];
}
s[K] = 1;
int delay = 0;
for(int i = 0;i < N - 1;i++){
int u = K,mind = 0x3f3f3f3f;
for(int j = 1;j <= N;j++){
if(!s[j] && dist[j] < mind){
u = j;
mind = dist[j];
}
}
s[u] = 1;
if(u == K) return -1;
delay = max(delay,mind);
for(int j = 1;j <= N;j++){
if(!s[j] && map[u][j] + dist[u] < dist[j])
dist[j] = map[u][j] + dist[u];
}
}
return delay;
}
};