-
Notifications
You must be signed in to change notification settings - Fork 154
/
the-maze-ii.js
110 lines (100 loc) · 3.14 KB
/
the-maze-ii.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/**
* The Maze II
*
* There is a ball in a maze with empty spaces and walls.
* The ball can go through empty spaces by rolling up, down, left or right,
* but it won't stop rolling until hitting a wall. When the ball stops, it could choose the next direction.
*
* Given the ball's start position, the destination and the maze,
* find the shortest distance for the ball to stop at the destination.
* The distance is defined by the number of empty spaces traveled by the ball
* from the start position (excluded) to the destination (included). If the ball
* cannot stop at the destination, return -1.
*
* The maze is represented by a binary 2D array. 1 means the wall and 0 means the empty space.
* You may assume that the borders of the maze are all walls.
* The start and destination coordinates are represented by row and column indexes.
*
* Example 1
*
* Input 1: a maze represented by a 2D array
*
* 0 0 1 0 0
* 0 0 0 0 0
* 0 0 0 1 0
* 1 1 0 1 1
* 0 0 0 0 0
*
* Input 2: start coordinate (rowStart, colStart) = (0, 4)
* Input 3: destination coordinate (rowDest, colDest) = (4, 4)
*
* Output: 12
* Explanation: One shortest way is : left -> down -> left -> down -> right -> down -> right.
* The total distance is 1 + 1 + 3 + 1 + 2 + 2 + 2 = 12.
*
* Example 2
*
* Input 1: a maze represented by a 2D array
*
* 0 0 1 0 0
* 0 0 0 0 0
* 0 0 0 1 0
* 1 1 0 1 1
* 0 0 0 0 0
*
* Input 2: start coordinate (rowStart, colStart) = (0, 4)
* Input 3: destination coordinate (rowDest, colDest) = (3, 2)
*
* Output: -1
* Explanation: There is no way for the ball to stop at the destination.
*
* Note:
*
* 1. There is only one ball and one destination in the maze.
* 2. Both the ball and the destination exist on an empty space,
* and they will not be at the same position initially.
* 3. The given maze does not contain border (like the red rectangle in the example pictures),
* but you could assume the border of the maze are all walls.
* 4. The maze contains at least 2 empty spaces, and both the width and height of the maze won't exceed 100.
*/
/**
* DFS Solution
*
* @param {number[][]} maze
* @param {number[]} start
* @param {number[]} destination
* @return {number}
*/
const shortestDistance = (maze, start, destination) => {
const m = maze.length;
const n = maze[0].length;
const distance = Array(m)
.fill()
.map(() => Array(n).fill(Number.MAX_SAFE_INTEGER));
distance[start[0]][start[1]] = 0;
dfs(maze, start, distance);
return distance[destination[0]][destination[1]] === Number.MAX_SAFE_INTEGER
? -1
: distance[destination[0]][destination[1]];
};
const dfs = (maze, start, distance) => {
const dirs = [[-1, 0], [1, 0], [0, -1], [0, 1]];
const [x, y] = start;
for (let [dx, dy] of dirs) {
let i = x + dx;
let j = y + dy;
let count = 0;
while (i >= 0 && j >= 0 && i < maze.length && j < maze[0].length && maze[i][j] === 0) {
count++;
i += dx;
j += dy;
}
i -= dx;
j -= dy;
if (distance[i][j] > distance[x][y] + count) {
distance[i][j] = distance[x][y] + count;
dfs(maze, [i, j], distance);
}
}
};
export { shortestDistance };