1. 程式人生 > >lintcode(115)不同的路徑2

lintcode(115)不同的路徑2

問題描述:

"不同的路徑" 的跟進問題:

現在考慮網格中有障礙物,那樣將會有多少條不同的路徑?

網格中的障礙和空位置分別用 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];
    }
}