Saturday, April 8, 2017

Unique Paths II

在Initialize的时候如果遇到一个obstacle的话,后面就都不能走了
所以可以直接break
for (i = 0; i < m; i++) {
            //System.out.println(obstacleGrid[i][0]);
            if (obstacleGrid[i][0] != 1) dp[i][0] = 1;
            else break;
        }

我自己写的一开始没有考虑这点
而且正确版本也不如break简洁

public class Solution {
    /**
     * @param obstacleGrid: A list of lists of integers
     * @return: An integer
     */
    public int uniquePathsWithObstacles(int[][] obstacleGrid) {
        // write your code here
        if (obstacleGrid == null || obstacleGrid.length == 0 || obstacleGrid[0] == null || obstacleGrid[0].length == 0) return -1;
        int m = obstacleGrid.length;
        int n = obstacleGrid[0].length;
        int[][] dp = new int[m][n];
        int i, j;
        dp[0][0] = obstacleGrid[0][0] == 1 ? 0 : 1;
        for (i = 1; i < m; i++) {
            if (obstacleGrid[i][0] == 1 || dp[i - 1][0] == 0) dp[i][0] = 0;
            else dp[i][0] = 1;
        }
        for (j = 1; j < n; j++) {
            if (obstacleGrid[0][j] == 1 || dp[0][j - 1] == 0) dp[0][j] = 0;
            else dp[0][j] = 1;
        }
        for (i = 1; i < m; i++) {
            for (j = 1; j < n; j++) {
                if (obstacleGrid[i][j] == 1) dp[i][j] = 0;
                else dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
            }
        }
        return dp[m - 1][n - 1];
    }
}

No comments:

Post a Comment