面试题 08.02. 迷路的机器人

题目

设想有个机器人坐在一个网格的左上角，网格 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。

解题思路

回溯

• 机器人可以向右或向下两个方向移动
• 如果遇到障碍就回到前一个网格，向另一个可移动方向移动
• 如果到达终点返回路径，否则返回空列表

代码

class Solution {
  private int row, column;
  private int[][] directions = {{0, 1}, {1, 0}};
  private List<List<Integer>> ans;

  public List<List<Integer>> pathWithObstacles(int[][] obstacleGrid) {
    ans = new ArrayList<>();
    row = obstacleGrid.length;
    column = obstacleGrid[0].length;
    backtrack(obstacleGrid, 0, 0, new ArrayList<>());
    return ans;
  }

  private void backtrack(int[][] obstacleGrid, int r, int c, List<List<Integer>> list) {
    if (r < 0 || c < 0 || r == row || c == column || obstacleGrid[r][c] == 1) return;
    list = new ArrayList<>(list);
    list.add(Arrays.asList(r, c));
    obstacleGrid[r][c] = 1;
    if (r == row - 1 && c == column - 1) {
      ans = list;
      return;
    }
    for (int[] dirc : directions) backtrack(obstacleGrid, r + dirc[0], c + dirc[1], list);
    list.remove(list.size() - 1);
  }
}

题目来源：力扣（LeetCode）