lintcode(115)不同的路徑2
阿新 • • 發佈:2018-12-26
問題描述:
"不同的路徑" 的跟進問題:
現在考慮網格中有障礙物,那樣將會有多少條不同的路徑?
網格中的障礙和空位置分別用 1 和 0 來表示。
示例:如下所示在3x3的網格中有一個障礙物:
[
[0,0,0],
[0,1,0],
[0,0,0]
]
一共有2條不同的路徑從左上角到右下角。
思路:二維陣列中,設定為1 的位置表示障礙,這意味著不能通過這個位置到達其右邊或下邊的位置,所以打到障礙位置的路徑數量可設定為0來佔位,這樣在計算下一個位置的時候,就排除這一點的可能性。
實現:
將第一行和第一列設定初始值為1,一旦遇到陣列原位置為1,則從該位置起至最後都設定為0(第一行的位只能通過左邊到達,一旦遇到障礙,則剩餘位置均不可達到,第一列與此相同)在計算其他位置時,要注意判斷,初始值是否為1,若為1則置0。
程式碼:
public class Solution { /** * @param obstacleGrid: A list of lists of integers * @return: An integer */ public int uniquePathsWithObstacles(int[][] obstacleGrid) { // write your code here int m = obstacleGrid.length; int n = obstacleGrid[0].length; if(obstacleGrid[0][0] == 1) return 0; for(int i = 0;i<m;i++){ if(obstacleGrid[i][0] == 1){ obstacleGrid[i][0] = 0; for(int k = i;k<m;k++){ obstacleGrid[k][0] = 0; } break; }else{ obstacleGrid[i][0] = 1; } } for(int j = 1;j<n;j++){ if(obstacleGrid[0][j] == 1){ obstacleGrid[0][j] = 0; for(int k = j;k<n;k++){ obstacleGrid[0][j] = 0; } break; }else{ obstacleGrid[0][j] = 1; } } for(int i = 1;i<m;i++){ for(int j = 1;j<n;j++){ if(obstacleGrid[i][j] == 1){ obstacleGrid[i][j] = 0; }else{ obstacleGrid[i][j] = obstacleGrid[i-1][j] + obstacleGrid[i][j-1]; } } } return obstacleGrid[m-1][n-1]; } }