diff --git a/graphs/dijkstra_binary_grid.py b/graphs/dijkstra_binary_grid.py index 06293a87da2d..736623a4cc2a 100644 --- a/graphs/dijkstra_binary_grid.py +++ b/graphs/dijkstra_binary_grid.py @@ -44,6 +44,9 @@ def dijkstra( (4.0, [(0, 0), (0, 1), (0, 2), (1, 2), (2, 2)]) """ rows, cols = grid.shape + if grid[source] != 1 or grid[destination] != 1: + return np.inf, [] + dx = [-1, 1, 0, 0] dy = [0, 0, -1, 1] if allow_diagonal: