题目:
给你一个二维数组 edges
表示一个 n
个点的无向图,其中 edges[i] = [ui, vi, lengthi]
表示节点 ui
和节点 vi
之间有一条需要 lengthi
单位时间通过的无向边。同时给你一个数组 disappear
,其中 disappear[i]
表示节点 i
从图中消失的时间点,在那一刻及以后,你无法再访问这个节点。注意,图有可能一开始是不连通的,两个节点之间也可能有多条边。
请你返回数组 answer
,answer[i]
表示从节点 0
到节点 i
需要的 最少 单位时间。如果从节点 0
出发 无法 到达节点 i
,那么 answer[i]
为 -1
。
示例 1:
输入:n = 3, edges = [[0,1,2],[1,2,1],[0,2,4]], disappear = [1,1,5]
输出:[0,-1,4]
解释:
我们从节点 0 出发,目的是用最少的时间在其他节点消失之前到达它们。
- 对于节点 0 ,我们不需要任何时间,因为它就是我们的起点。
- 对于节点 1 ,我们需要至少 2 单位时间,通过
edges[0]
到达。但当我们到达的时候,它已经消失了,所以我们无法到达它。 - 对于节点 2 ,我们需要至少 4 单位时间,通过
edges[2]
到达。
示例 2:
输入:n = 3, edges = [[0,1,2],[1,2,1],[0,2,4]], disappear = [1,3,5]
输出:[0,2,3]
解释:
我们从节点 0 出发,目的是用最少的时间在其他节点消失之前到达它们。
- 对于节点 0 ,我们不需要任何时间,因为它就是我们的起点。
- 对于节点 1 ,我们需要至少 2 单位时间,通过
edges[0]
到达。 - 对于节点 2 ,我们需要至少 3 单位时间,通过
edges[0]
和edges[1]
到达。
示例 3:
输入:n = 2, edges = [[0,1,1]], disappear = [1,1]
输出:[0,-1]
解释:
当我们到达节点 1 的时候,它恰好消失,所以我们无法到达节点 1 。
提示:
1 <= n <= 5 * 104
0 <= edges.length <= 105
edges[i] == [ui, vi, lengthi]
0 <= ui, vi <= n - 1
1 <= lengthi <= 105
disappear.length == n
1 <= disappear[i] <= 105
思路:
对于本题,answer 几乎就是 dis 数组。只需要在 Dijkstra 算法的过程中,添加一个判断:在更新最短路之前,如果最短路长度 ≥ disappear[i],说明无法及时到达节点 i,不更新。
问:我是这样做的,先不管 disappear,用 Dijkstra 算法模板求出 dis 数组,然后把其中 dis[i] ≥ disappear[i] 的 dis[i] 改成 −1。这个做法是否正确?
答:这个做法是错的。考虑这样的数据:离节点 0 近的节点 x,其 disappear[x] 很小;离节点 0 远的节点 y,其 disappear[y] 很大。由于 disappear[x] 很小,我们无法通过节点 x,所以远处的节点 y 实际上也无法到达。但错误做法会因为 dis[y] < disappear[y],误认为节点 y 可以到达,最终返回错误的答案。
代码:
class Solution {
public:
vector<int> minimumTime(int n, vector<vector<int>>& edges, vector<int>& disappear) {
// 构建领接表
vector<vector<pair<int, int>>> g(n);
for (auto& e : edges) {
int x = e[0], y = e[1], wt = e[2];
g[x].emplace_back(y, wt);
g[y].emplace_back(x, wt);
}
// 初始化
vector<int> dis(n, -1);
dis[0] = 0;
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<>> pq; // pair的降序
pq.emplace(0, 0);
// Dijkstra算法求最短路
while (!pq.empty()) {
auto [dx, x] = pq.top();
pq.pop();
if (dx > dis[x]) { // x 之前出过堆
continue;
}
for (auto& [y, d] : g[x]) {
int new_dis = dx + d;
if (new_dis < disappear[y] && (dis[y] < 0 || new_dis < dis[y])) {
dis[y] = new_dis;
pq.emplace(new_dis, y);
}
}
}
return dis;
}
};