设想有个机器人坐在一个网格的左上角,网格 r 行 c 列。机器人只能向下或向右移动,但不能走到一些被禁止的网格(有障碍物)。设计一种算法,寻找机器人从左上角移动到右下角的路径。
网格中的障碍物和空位置分别用 1
和 0
来表示。
返回一条可行的路径,路径由经过的网格的行号和列号组成。左上角为 0 行 0 列。
示例 1:
输入: [ [0,0,0], [0,1,0], [0,0,0] ] 输出: [[0,0],[0,1],[0,2],[1,2],[2,2]] 解释: 输入中标粗的位置即为输出表示的路径,即 0行0列(左上角) -> 0行1列 -> 0行2列 -> 1行2列 -> 2行2列(右下角)
说明:r 和 c 的值均不超过 100。
DFS 思路
剪枝:
防止多次进入同一个位置,走过的位置要将其置为 1
function pathWithObstacles(obstacleGrid: number[][]): number[][] {
const m = obstacleGrid.length;
const n = obstacleGrid[0].length;
const res = [];
const dfs = (i: number, j: number): boolean => {
if (i === m || j === n || obstacleGrid[i][j] === 1) {
return false;
}
res.push([i, j]);
obstacleGrid[i][j] = 1;
if ((i + 1 === m && j + 1 === n) || dfs(i + 1, j) || dfs(i, j + 1)) {
return true;
}
res.pop();
return false;
};
if (dfs(0, 0)) {
return res;
}
return [];
}
impl Solution {
fn dfs(grid: &mut Vec<Vec<i32>>, path: &mut Vec<Vec<i32>>, i: usize, j: usize) -> bool {
if i == grid.len() || j == grid[0].len() || grid[i][j] == 1 {
return false;
}
path.push(vec![i as i32, j as i32]);
grid[i as usize][j as usize] = 1;
if (i + 1 == grid.len() && j + 1 == grid[0].len())
|| Self::dfs(grid, path, i + 1, j)
|| Self::dfs(grid, path, i, j + 1)
{
return true;
}
path.pop();
false
}
pub fn path_with_obstacles(mut obstacle_grid: Vec<Vec<i32>>) -> Vec<Vec<i32>> {
let mut res = vec![];
if Self::dfs(&mut obstacle_grid, &mut res, 0, 0) {
return res;
}
vec![]
}
}