This is a fundamental DP problem. First of all, let's make some observations.
Since the robot can only move right and down, when it arrives at a point, there are only two possibilities:
- It arrives at that point from above (moving down to that point);
- It arrives at that point from left (moving right to that point).
Thus, we have the following state equations: suppose the number of paths to arrive at a point (i, j)
is denoted as P[i][j]
, it is easily concluded that P[i][j] = P[i - 1][j] + P[i][j - 1]
.
The boundary conditions of the above equation occur at the leftmost column (P[i][j - 1]
does not exist) and the uppermost row (P[i - 1][j]
does not exist). These conditions can be handled by initialization (pre-processing) --- initialize P[0][j] = 1, P[i][0] = 1
for all valid i, j
. Note the initial value is 1
instead of 0
!
Now we can write down the following (unoptimized) code.
1 class Solution { 2 int uniquePaths(int m, int n) { 3 vector<vector<int> > path(m, vector<int> (n, 1)); 4 for (int i = 1; i < m; i++) 5 for (int j = 1; j < n; j++) 6 path[i][j] = path[i - 1][j] + path[i][j - 1]; 7 return path[m - 1][n - 1]; 8 } 9 };
As can be seen, the above solution runs in O(n^2)
time and costs O(m*n)
space. However, you may have observed that each time when we update path[i][j]
, we only need path[i - 1][j]
(at the same column) and path[i][j - 1]
(at the left column). So it is enough to maintain two columns (the current column and the left column) instead of maintaining the full m*n
matrix. Now the code can be optimized to have O(min(m, n))
space complexity.
1 class Solution { 2 int uniquePaths(int m, int n) { 3 if (m > n) return uniquePaths(n, m); 4 vector<int> pre(m, 1); 5 vector<int> cur(m, 1); 6 for (int j = 1; j < n; j++) { 7 for (int i = 1; i < m; i++) 8 cur[i] = cur[i - 1] + pre[i]; 9 swap(pre, cur); 10 } 11 return pre[m - 1]; 12 } 13 };
Further inspecting the above code, we find that keeping two columns is used to recover pre[i]
, which is just cur[i]
before its update. So there is even no need to use two vectors and one is just enough. Now the space is further saved and the code also gets much shorter.
1 class Solution { 2 int uniquePaths(int m, int n) { 3 if (m > n) return uniquePaths(n, m); 4 vector<int> cur(m, 1); 5 for (int j = 1; j < n; j++) 6 for (int i = 1; i < m; i++) 7 cur[i] += cur[i - 1]; 8 return cur[m - 1]; 9 } 10 };
Well, till now, I guess you may even want to optimize it to O(1)
space complexity since the above code seems to rely on only cur[i]
and cur[i - 1]
. You may think that 2 variables is enough? Well, it is not. Since the whole cur
needs to be updated for n - 1
times, it means that all of its values need to be saved for next update and so two variables is not enough.