来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/network-delay-time
有 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) 对都 互不相同(即,不含重复边)
思路:简单的bfs,就把给出来的边的条件转换成边集数组那一套,直接bfs就可以啦。比较简单。
class Solution {
public:
int i, ans, k1, j, length;
bool f[1000];
int q[1000], v[1000];
int point[7000], next[7000], first[1000], w[7000];
void bfs(int s, int n)
{
int head = 0, tail = 1, k;
for (i = 1; i <= n; i++)
f[i] = false;
f[s] = true;
for (i = 1; i <= n; i++){
v[i] = 100000000;
}
v[s] = 0;
q[1] = s;
while (head < tail){
head++;
j = q[head];
f[j] =false;
k = first[j];
while (k != 0){
if (v[j] + w[k] < v[point[k]]){
// ans = 555;
v[point[k]] = v[j] + w[k];
if (f[point[k]] == false){
f[point[k]] = true;
tail++;
q[tail] = point[k];
}
}
k = next[k];
}
}
}
void add(int s, int t, int w1)
{
k1++;
point[k1] = t;next[k1] = first[s];first[s] = k1;w[k1] = w1;
}
int networkDelayTime(vector<vector<int>>& times, int n, int k) {
length = times.size();
for (i = 0; i < length; i++){
add(times[i][0],times[i][1], times[i][2]);
}
bfs(k, n);
// return ans;
for (i = 1; i <= n; i++)
ans = max(ans, v[i]);
if (ans != 100000000)
return ans;
return -1;
}
};