来源:力扣(LeetCode)
描述:
在一个有向图中,节点分别标记为 0, 1, ..., n-1
。图中每条边为红色或者蓝色,且存在自环或平行边。
red_edges
中的每一个 [i, j]
对表示从节点 i
到节点 j
的红色有向边。类似地,blue_edges
中的每一个 [i, j]
对表示从节点 i
到节点 j
的蓝色有向边。
返回长度为 n
的数组 answer
,其中 answer[X]
是从节点 0
到节点 X
的红色边和蓝色边交替出现的最短路径的长度。如果不存在这样的路径,那么 answer[x] = -1
。
示例 1:
输入:n = 3, red_edges = [[0,1],[1,2]], blue_edges = []
输出:[0,1,-1]
示例 2:
输入:n = 3, red_edges = [[0,1]], blue_edges = [[2,1]]
输出:[0,1,-1]
示例 3:
输入:n = 3, red_edges = [[1,0]], blue_edges = [[2,1]]
输出:[0,-1,-1]
示例 4:
输入:n = 3, red_edges = [[0,1]], blue_edges = [[1,2]]
输出:[0,1,2]
示例 5:
输入:n = 3, red_edges = [[0,1],[0,2]], blue_edges = [[1,0]]
输出:[0,1,1]
提示:
- 1 <= n <= 100
- red_edges.length <= 400
- blue_edges.length <= 400
- red_edges[i].length == blue_edges[i].length == 2
- 0 <= red_edges[i][j], blue_edges[i][j] < n
方法:广度优先搜索
使用 0 表示红色,1 表示蓝色,对于某一节点 x,从节点 0 到节点 x 的红色边和蓝色边交替出现的路径有两种类型:
- 类型 0:路径最终到节点 x 的有向边为红色;
- 类型 1:路径最终到节点 x 的有向边为蓝色。
如果存在从节点 0 到节点 x 的类型 0 的颜色交替路径,并且从节点 x 到节点 y 的有向边为蓝色,那么该路径加上该有向边组成了从节点 0 到节点 y 的类型 1 的颜色交替路径。类似地,如果存在从节点 0 到节点 x 的类型 1 的颜色交替路径,并且从节点 x 到节点 y 的有向边为红色,那么该路径加上该有向边组成了从节点 0 到节点 y 的类型 0 的颜色交替路径。
使用广度优先搜索获取从节点 0 到某一节点的两种类型的颜色交替最短路径的长度,广度优先搜索的队列元素由节点编号和节点路径类型组成,初始时节点 0 到节点 0 的两种类型的颜色交替最短路径的长度都是 0,将两个初始值入队。对于某一个队列元素,节点编号为 x,节点路径类型为 t,那么根据类型 t 选择颜色为 1 − t 的相邻有向边,如果有向边的终点节点 y 对应类型 1 − t 没有被访问过,那么更新节点 y 的类型 1 − t 的颜色交替最短路径的长度为节点 x 的类型 t 的颜色交替最短路径的长度加 1,并且将它入队。
从节点 0 到某一节点的颜色交替最短路径的长度为两种类型的颜色交替最短路径长度的最小值。
代码:
class Solution {
public:
vector<int> shortestAlternatingPaths(int n, vector<vector<int>>& redEdges, vector<vector<int>>& blueEdges) {
vector<vector<vector<int>>> next(2, vector<vector<int>>(n));
for (auto &e : redEdges) {
next[0][e[0]].push_back(e[1]);
}
for (auto &e : blueEdges) {
next[1][e[0]].push_back(e[1]);
}
vector<vector<int>> dist(2, vector<int>(n, INT_MAX)); // 两种类型的颜色最短路径的长度
queue<pair<int, int>> q;
dist[0][0] = 0;
dist[1][0] = 0;
q.push({0, 0});
q.push({0, 1});
while (!q.empty()) {
auto [x, t] = q.front();
q.pop();
for (auto y : next[1 - t][x]) {
if (dist[1 - t][y] != INT_MAX) {
continue;
}
dist[1 - t][y] = dist[t][x] + 1;
q.push({y, 1 - t});
}
}
vector<int> answer(n);
for (int i = 0; i < n; i++) {
answer[i] = min(dist[0][i], dist[1][i]);
if (answer[i] == INT_MAX) {
answer[i] = -1;
}
}
return answer;
}
};
执行用时:12 ms, 在所有 C++ 提交中击败了94.70%的用户
内存消耗:14.3 MB, 在所有 C++ 提交中击败了81.13%的用户
复杂度分析
时间复杂度:O(n+r+b),其中 n 是节点数,r 是红色边的数目,b 是蓝色边的数目。广度优先搜索最多访问一个节点两次,最多访问一条边一次,因此时间复杂度为 O(n+r+b)。
空间复杂度:O(n+r+b)。队列中最多有 2n 个元素,保存 next 需要 O(r+b) 的空间,保存 dist 需要 O(n) 的空间。
author:LeetCode-Solution