063J. Unique Paths II

https://leetcode.com/problems/unique-paths-ii/

Method DP

这个题和上一题的观察一样。但是不同的是,有障碍. 首先我们让遇到障碍的时候,它的值变成0。这样做动态规划的时候, 从山这一侧的道路总数就变成了0. 其次我们让路线数量变为负数。这样就和山区别开来了。

class Solution {
    public int uniquePathsWithObstacles(int[][] obstacleGrid) {
        int m = obstacleGrid.length;
        int n = obstacleGrid[0].length;
        if(obstacleGrid[0][0] == 1) return 0;
        obstacleGrid[0][0] = -1;
        for(int i = 1; i < m; i++){
            if(obstacleGrid[i][0] == 1){
                obstacleGrid[i][0] = 0;                    
                continue;
            }            
             obstacleGrid[i][0] = obstacleGrid[i-1][0];
        }
        for(int i = 1; i < n; i++){
            if(obstacleGrid[0][i] == 1){
                obstacleGrid[0][i] = 0;                    
                continue;
            }
             obstacleGrid[0][i] = obstacleGrid[0][i-1];
        }        
        for(int i = 1; i < m; i++){
            for(int j = 1; j < n; j++){
                if(obstacleGrid[i][j] == 1){
                    obstacleGrid[i][j] = 0;                    
                    continue;
                } 
                obstacleGrid[i][j] = obstacleGrid[i-1][j]+obstacleGrid[i][j-1];
            }
        }
        return -obstacleGrid[m-1][n-1];
    }
}

上面的代码被优化合并成了如下代码。

class Solution {
    public int uniquePathsWithObstacles(int[][] obstacleGrid) {
        int m = obstacleGrid.length;
        int n = obstacleGrid[0].length;
 
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(obstacleGrid[i][j] == 1){
                    obstacleGrid[i][j] = 0;                    
                    continue;
                }
                if(i < 1 && j < 1){
                   obstacleGrid[0][0] = -1;   
                }else if(i < 1){
                   obstacleGrid[0][j] = obstacleGrid[0][j-1];
                }else if( j < 1){
                   obstacleGrid[i][0] = obstacleGrid[i-1][0];  
                }else{
                   obstacleGrid[i][j] = obstacleGrid[i-1][j]+obstacleGrid[i][j-1]; 
                }
            }
        }
        return -obstacleGrid[m-1][n-1];
    }
}

下面是一个2023的代码: 感觉之前写的多少有点 over-engineering.

class Solution {
    public int uniquePathsWithObstacles(int[][] obstacleGrid) {
        int m = obstacleGrid.length;
        int n = obstacleGrid[0].length;
        int[][] dp = new int[m][n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (obstacleGrid[i][j] == 1) {
                    dp[i][j] = 0;
                    continue;
                }
                if (i == 0 && j == 0) {
                    dp[i][j] = 1;
                } else if (i == 0) {
                    dp[i][j] = dp[i][j - 1];
                } else if (j == 0) {
                    dp[i][j] = dp[i - 1][j];
                } else {
                    dp[i][j] = dp[i][j - 1] + dp[i - 1][j];
                }
            }
        }
        return dp[m - 1][n - 1];
    }
}

Last updated

Was this helpful?