[leetcode] 63. Unique Paths II (medium)

版权声明:by ruohua3kou https://blog.csdn.net/ruohua3kou/article/details/83098811

原题
思路:
用到dp的思想,到row,col点路径数量 :

path[row][col]=path[row][col-1]+path[row-1][col];

遍历row*col,如果map[row][col]为1,则将其置为0;如果非1,则进行上述公式。
最后返回path[终点row][终点col]的值即为解。

一开始的代码,beat 44%,效率不高

class Solution {
 public:
  int uniquePathsWithObstacles(vector<vector<int>> &obstacleGrid) {
    int rowSize = obstacleGrid.size();
    int colSize = obstacleGrid[0].size();

    if (obstacleGrid[0][0] == 1) return 0;

    obstacleGrid[0][0] = 1;
    for (int col = 1; col < colSize; col++) {
      if (obstacleGrid[0][col] == 1)
        obstacleGrid[0][col] = 0;
      else
        obstacleGrid[0][col] += obstacleGrid[0][col - 1];
    }
    for (int row = 1; row < rowSize; row++) {
      if (obstacleGrid[row][0] == 1)
        obstacleGrid[row][0] = 0;
      else
        obstacleGrid[row][0] += obstacleGrid[row-1][0];
    }
    for (int i = 1; i < rowSize; i++) {
      for (int j = 1; j < colSize; j++) {
        if (obstacleGrid[i][j] == 1)
          obstacleGrid[i][j] = 0;
        else
          obstacleGrid[i][j] = obstacleGrid[i - 1][j] + obstacleGrid[i][j - 1];
      }
    }
    return obstacleGrid[rowSize - 1][colSize - 1];
  }
};

优化后的代码:
使用额外一个数组记录走到每一列有几种走法,因为题目只求终点,则使用一维数组即可。
使用一个额外的整型pre记录当前的前一步有多少种走法。(左边走来+上边走来)
则有:

pre(当前)=dp[col](上一行当前列) + pre(左一格当前行);
dp[col](当前)=pre;

beat 100%

class Solution {
 public:
  int uniquePathsWithObstacles(vector<vector<int>> &obstacleGrid) {
    int rowSize = obstacleGrid.size();
    int colSize = obstacleGrid[0].size();
    int dp[colSize] = {0};
    int pre = 1;
    for (int i = 0; i < rowSize; i++) {
      for (int j = 0; j < colSize; j++) {
        if (obstacleGrid[i][j] == 0) {
          pre += dp[j];
          dp[j] = pre;
        }
        else {
          pre = 0;
          dp[j] = 0;
        }
      }
      pre=0;
    }
    return dp[colSize-1];
  }
};

猜你喜欢

转载自blog.csdn.net/ruohua3kou/article/details/83098811
今日推荐