题目描述(中等难度)

63. Unique Paths II - 图1

62题的变体,增加了一些不能走的格子,用 1 表示。还是输出从左上角到右下角总共有多少种走法。

没做过62题的话可以先看一下,62 题总结的很详细了,我直接在 62 题的基础上改了。

解法一 递归

  1. public int uniquePathsWithObstacles(int[][] obstacleGrid) {
  2. int m = obstacleGrid.length;
  3. int n = obstacleGrid[0].length;
  4. HashMap<String, Integer> visited = new HashMap<>();
  5. //起点是障碍,直接返回 0
  6. if (obstacleGrid[0][0] == 1)
  7. return 0;
  8. return getAns(0, 0, m - 1, n - 1, 0, visited, obstacleGrid);
  9. }
  10. private int getAns(int x, int y, int m, int n, int num, HashMap<String, Integer> visited, int[][] obstacleGrid) {
  11. // TODO Auto-generated method stub
  12. if (x == m && y == n) {
  13. return 1;
  14. }
  15. int n1 = 0;
  16. int n2 = 0;
  17. String key = x + 1 + "@" + y;
  18. if (!visited.containsKey(key)) {
  19. //与 62 题不同的地方,增加了判断是否是障碍
  20. if (x + 1 <= m && obstacleGrid[x + 1][y] == 0) {
  21. n1 = getAns(x + 1, y, m, n, num, visited, obstacleGrid);
  22. }
  23. } else {
  24. n1 = visited.get(key);
  25. }
  26. key = x + "@" + (y + 1);
  27. if (!visited.containsKey(key)) {
  28. //与 62 题不同的地方,增加了判断是否是障碍
  29. if (y + 1 <= n && obstacleGrid[x][y + 1] == 0) {
  30. n2 = getAns(x, y + 1, m, n, num, visited, obstacleGrid);
  31. }
  32. } else {
  33. n2 = visited.get(key);
  34. }
  35. //将当前点加入 visited 中
  36. key = x + "@" + y;
  37. visited.put(key, n1+n2);
  38. return n1 + n2;
  39. }

时间复杂度:

空间复杂度:

解法二 动态规划

62题解法二最后个想法上改。

  1. public int uniquePathsWithObstacles(int[][] obstacleGrid) {
  2. int m = obstacleGrid.length;
  3. int n = obstacleGrid[0].length;
  4. //起点是障碍,直接返回 0
  5. if (obstacleGrid[0][0] == 1)
  6. return 0;
  7. int[] dp = new int[n];
  8. int i = 0;
  9. //初始化第一行和 62 题不一样了
  10. //这里的话不是全部初始化 1 了,因为如果有障碍的话当前列和后边的列就都走不过了,初始化为 0
  11. for (; i < n; i++) {
  12. if (obstacleGrid[0][i] == 1) {
  13. dp[i] = 0;
  14. break;
  15. } else {
  16. dp[i] = 1;
  17. }
  18. }
  19. //障碍后边的列全部初始化为 0
  20. for (; i < n; i++) {
  21. dp[i] = 0;
  22. }
  23. for (i = 1; i < m; i++) {
  24. //这里改为从 0 列开始了,因为 62 题中从 1 开始是因为第 0 列永远是 1 不需要更新
  25. //这里的话,如果第 0 列是障碍的话,需要更新为 0
  26. for (int j = 0; j < n; j++) {
  27. if (obstacleGrid[i][j] == 1) {
  28. dp[j] = 0;
  29. } else {
  30. //由于从第 0 列开始了,更新公式有 j - 1,所以 j = 0 的时候不更新
  31. if (j != 0)
  32. dp[j] = dp[j] + dp[j - 1];
  33. }
  34. }
  35. }
  36. return dp[(n - 1)];
  37. }

时间复杂度:O(m * n)。

空间复杂度:O(n)。

和 62 题改动不大,就是在障碍的地方,更新的时候需要注意一下。

windliang wechat

添加好友一起进步~

如果觉得有帮助的话,可以点击 这里 给一个 star 哦 ^^

如果想系统的学习数据结构和算法,强烈推荐一个我之前学过的课程,可以点击 这里 查看详情