leetcode 63. Unique Paths II
阿新 • • 發佈:2018-12-26
前面做過一道Unique Paths的題,那就順便把Unique Paths II做了吧。這道題的要求基本和前面相同,從左上角出發,只能向左或者向右,求最終到達右下角有多少條路徑,區別是這題多了障礙點,就是矩陣上數值為1的地方,有障礙的地方無法通行。
為了求解這道題,採取和之前類似的做法,先確定第一行和第一列的路徑數,如果沒遇到障礙那就全是1,如果有障礙,那麼從障礙開始全是0。求出第一行和第一列的路徑數之後,從左到右從上到下,求其他點的路徑數,如果原先它在矩陣上的值是1,那麼它的路徑數為0。最終得到起點到終點的路徑數。
程式碼如下:
class Solution {
public :
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m = obstacleGrid.size();
int n = obstacleGrid[0].size();
if (m == 0||n == 0) return 0;
bool firstObstacle = false; // 第一格是不是障礙
bool hasObstacleFront = false;
if (obstacleGrid[0 ][0] == 1) firstObstacle = true;
// 判斷第一行的路徑數
for (int i = 0; i < n; i++) {
if (obstacleGrid[0][i] == 1) {
hasObstacleFront = true;
}
if (hasObstacleFront) {
for (int j = i; j < n; j++) {
obstacleGrid[0 ][j] = 0;
}
break;
} else {
obstacleGrid[0][i] = 1;
}
}
// 第一列的路徑數
hasObstacleFront = firstObstacle;
for (int i = 1; i < m; i++) {
if (obstacleGrid[i][0] == 1) {
hasObstacleFront = true;
}
if (hasObstacleFront) {
for (int j = i; j < m; j++) {
obstacleGrid[j][0] = 0;
}
break;
} else {
obstacleGrid[i][0] = 1;
}
}
// 其他位置的路徑數
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (obstacleGrid[i][j]) {
obstacleGrid[i][j] = 0;
} else {
obstacleGrid[i][j] = obstacleGrid[i - 1][j] + obstacleGrid[i][j - 1];
}
}
}
return obstacleGrid[m - 1][n - 1];
}
};