You are given an integer n
, the number of nodes in a directed graph where the nodes are labeled from 0
to n - 1
. Each edge is red or blue in this graph, and there could be self-edges and parallel edges.
You are given two arrays redEdges
and blueEdges
where:
redEdges[i] = [ai, bi]
indicates that there is a directed red edge from node ai
to node bi
in the graph, andblueEdges[j] = [uj, vj]
indicates that there is a directed blue edge from node uj
to node vj
in the graph.Return an array answer
of length n
, where each answer[x]
is the length of the shortest path from node 0
to node x
such that the edge colors alternate along the path, or -1
if such a path does not exist.
Example 1:
Input: n = 3, redEdges = [[0,1],[1,2]], blueEdges = [] Output: [0,1,-1]
Example 2:
Input: n = 3, redEdges = [[0,1]], blueEdges = [[2,1]] Output: [0,1,-1]
Constraints:
1 <= n <= 100
0 <= redEdges.length, blueEdges.length <= 400
redEdges[i].length == blueEdges[j].length == 2
0 <= ai, bi, uj, vj < n
This problem asks for the shortest path between node 0 and all other nodes in a directed graph, with the constraint that the colors of the edges must alternate (red, blue, red, ... or blue, red, blue, ...). We can efficiently solve this using Breadth-First Search (BFS).
Approach:
Graph Representation: We represent the graph using two adjacency lists: one for red edges and one for blue edges. This allows us to easily access the neighbors of a node for each color.
BFS with Color Tracking: The BFS algorithm is modified to track the current edge color. Each node in the queue is represented as a tuple (node, color)
, where node
is the node's index and color
indicates the color of the last edge traversed (0 for red, 1 for blue).
Alternating Colors: When processing a node, we explore its neighbors using the opposite color. If the current color is red (0), we explore blue edges; if blue (1), we explore red edges. This ensures the alternating color constraint is met.
Shortest Path Distance: The distance from node 0 to each node is determined by the level in the BFS traversal. We use an array ans
to store the shortest distance for each node, initializing it with -1 to indicate that the node is unreachable.
Visited Tracking: A vis
set (or 2D boolean array) is crucial to avoid cycles and ensure we only visit each node with a specific color once. The key is (node, color)
, preventing revisits with the same color.
Time Complexity: O(V + E), where V is the number of nodes (n) and E is the number of edges (total number of red and blue edges). The BFS algorithm visits each node and edge at most once.
Space Complexity: O(V + E), dominated by the adjacency lists and the queue in the worst case (when the graph is dense).
Code Implementation (Python):
from collections import defaultdict, deque
class Solution:
def shortestAlternatingPaths(self, n: int, redEdges: List[List[int]], blueEdges: List[List[int]]) -> List[int]:
g = [defaultdict(list), defaultdict(list)] # Adjacency lists for red (0) and blue (1) edges
for u, v in redEdges:
g[0][u].append(v)
for u, v in blueEdges:
g[1][u].append(v)
ans = [-1] * n # Shortest distance to each node
ans[0] = 0 # Distance to starting node is 0
vis = set() # Visited nodes with colors
q = deque([(0, 0), (0, 1)]) # Queue of (node, color) tuples
vis.add((0,0))
vis.add((0,1))
d = 0 # Level/Distance
while q:
for _ in range(len(q)): # Process nodes at current level
i, c = q.popleft()
# Explore neighbors with opposite color
next_color = 1 - c
for neighbor in g[next_color][i]:
if (neighbor, next_color) not in vis:
vis.add((neighbor, next_color))
q.append((neighbor, next_color))
ans[neighbor] = d + 1
d += 1 #Increment distance for next level
return ans
The code in other languages (Java, C++, Go, TypeScript) follows a similar structure, adapting the data structures and syntax accordingly. The core algorithm remains the same: BFS with color tracking and efficient visited node management.