求一个机器人从(0,0)走到(m,n)的路径数量,只能往下或者往右移动一步。
f(i,j)表示从(0,0)走到(i,j)的路径数量,则f(i,j)=f(i-1,j)+f(i,j-1),画图可以很轻松的看出怎么求每个位置和初始赋值,第一行和第一列全都是1,然后求出即可。
这样的时间复杂度为O(mn),空间复杂度为O(mn)
class Solution {
public int uniquePaths(int m, int n) {
int[][] dp=new int[m][n];
for(int i=0;i<m;i++){
dp[i][0]=1;
}
for(int i=0;i<n;i++){
dp[0][i]=1;
}
for(int i=1;i<m;i++){
for(int j=1;j<n;j++){
dp[i][j]=dp[i-1][j]+dp[i][j-1];
}
}
return dp[m-1][n-1];
}
}
动态规划优化可以从空间复杂度来入手。
因为每次的循环中,所有的数只用了一次,所以在用完这两个数之后可以用下个数来代替。
创建一个n长的一维数组,在循环中,用dp[i]来表示dp[i][j-1],用dp[i-1]来表示dp[i-1][j],则这两个加完之后再赋值给dp[i]。
class Solution {
public int uniquePaths(int m, int n) {
int[] dp=new int[n];
for(int i=0;i<n;i++){
dp[i]=1;
}
for(int i=1;i<m;i++){
for(int j=1;j<n;j++){
dp[j]=dp[j-1]+dp[j];
}
}
return dp[n-1];
}
}