在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