63. Unique Paths II

1.問題

  • 計算從起點到終點有多少可能的途徑

  • 但中間有障礙物

2.想法

  • 提問

  • function header, parameter

  • test input

  • 說明想法

    • 動態規劃, 到每一點的途徑等於前一點(上面, 左邊)的總和

    • 特例: 因為機器人只能往下面跟往右邊, 因此對最上方的row來說, 不會有來自上方的機器人, 對最左邊的column來說, 不會有來自左邊的機器人

    • 當該點有障礙物時, 該點的值應為0, 表示無法抵達該點

  • 測試計算複雜度

3.程式碼

class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
        int m = obstacleGrid.size();
        int n = obstacleGrid[0].size();
        
        if (m == 0 || n == 0) {
            return 0;
        }
        
        //每個值存放的是到這邊所可能的路徑數
        vector<vector<int>> dp(m, vector<int>(n, 0));
        dp[0][0] = 1;
        
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (obstacleGrid[i][j] == 1) {
                    dp[i][j] = 0;
                } else if (i == 0 && j > 0) {
                    //left
                    dp[i][j] = dp[i][j - 1]; 
                } else if (i > 0 && j == 0) {
                    //top
                    dp[i][j] = dp[i - 1][j];
                } else if (i > 0 && j > 0){
                   dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
                }
            }
        }
        
        return dp[m - 1][n - 1];
    }
};

Last updated