1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
| #include <iostream> #include <vector> using namespace std;
class Solution { public: int uniquePathsWithObstacles(vector<vector<int>>& grid) { int m = grid.size(), n = grid[0].size(); if (grid[0][0] == 1 || grid[m-1][n-1] == 1) { return 0; }
vector<vector<int>> f(m+1, vector<int>(n+1, 0)); f[1][1] = grid[0][0] == 1 ? 0 : 1;
for (int i = 1; i <= m; i++) { for (int j = 1; j <= n; j++) { if (grid[i-1][j-1] != 0) { f[i][j] = 0; continue; } else { if (i == 1 && j == 1) { continue; } f[i][j] = f[i-1][j] + f[i][j-1]; } } }
return f[m][n]; } };
class Solution_2 { public: int uniquePathsWithObstacles(vector<vector<int>>& grid) { int m = grid.size(), n = grid[0].size(); if (grid[0][0] == 1 || grid[m-1][n-1] == 1) { return 0; }
vector<int> f(n+1, 0); f[1] = grid[0][0] == 1 ? 0 : 1;
for (int i = 1; i <= m; i++) { for (int j = 1; j <= n; j++) { if (grid[i-1][j-1] != 0) { f[j] = 0; } else { if (i == 1 && j == 1) { continue; } f[j] = f[j] + f[j-1]; } } }
return f[n]; } };
void printArray(const vector<vector<int>>& nums) { cout << "["; for (size_t i = 0; i < nums.size(); i++) { cout << "["; for (size_t j = 0; j < nums[i].size(); j++) { cout << nums[i][j]; if (j != nums[i].size() - 1) cout << ","; } cout << "]"; if (i != nums.size() - 1) cout << ","; } cout << "]" << endl; }
int main() { Solution solution; vector<vector<vector<int>>> grid_cases = { {{0,0,0},{0,1,0},{0,0,0}}, {{0,1},{0,0}} };
for (int i = 0; i < grid_cases.size(); i++) { auto& grid = grid_cases[i]; int result = solution.uniquePathsWithObstacles(grid);
cout << "Input: grid = "; printArray(grid); cout << "Output: " << result << endl; }
return 0; }
|