题目描述:
题解:
很经典的动态规划题,令f(i,j)为从(1,1)走到地图中(i,j)位置时所有可能的路径数。则机器人走到(i,j)这个位置的上一步可能是从上方的一个格子(i-1,j)往下走一步而来,也有可能是从左边的一个格子(i,j-1)而来,所以路径之和的问题分为这两个子问题,状态方程为:
f(i,j) = f(i-1,j)+f(i,j-1)
特别地,由于当处于第一行和第一列时,即i=1和j=1时,机器人只能通过向右走到达(1,j)和向下走到达(i,1)。所以初始化f(i,1)=1和f(1,j)=1
二维数组的动态规划代码为:
class Solution {
public int uniquePaths(int m, int n) {
int[][] map = new int[m+1][n+1];
for(int i=0;i<=m;i++) {
map[i][0]=0;
}
for(int j=0;j<=n;j++) {
map[0][j]=0;
}
for(int i=1;i<=m;i++) {
map[i][1]=1;
}
for(int j=1;j<=n;j++) {
map[1][j]=1;
}
for(int i=2;i<=m;i++) {
for(int j=2;j<=n;j++) {
map[i][j]=map[i-1][j]+map[i][j-1];
}
}
return map[m][n];
}
}
执行结果:
注意到:
我们每次计算f(i,j)时,只用到了第i行的数据和第i-1行的数据,那么可以只用两个一维数组进行保存进行空间的优化,每一行的数据填写完成后,数据更新。代码为:
class Solution {
public int uniquePaths(int m, int n) {
int cur[] = new int[n+1];//当前行
int pre[] = new int[n+1];//当前的行上一行
Arrays.fill(cur, 1);
Arrays.fill(pre,1);
//每次只保存两行,所以用两个单独的一维数组即可
for(int i=2;i<=m;i++) {
for(int j=2;j<=n;j++) {
cur[j]=cur[j-1]+pre[j];
}
pre=cur.clone();//当前行的数据更新完以后 把pre更新为当前行,然后再从下一行扫描
}
return pre[n];
}
}
执行结果:
再次进行优化,只用一个一维数组进行数据的更新,代码为:
class Solution {
//解法三:只用一个一维数组
public int uniquePaths(int m, int n) {
int[] cur = new int[n+1];
Arrays.fill(cur,1);
for (int i = 2; i <= m;i++){
for (int j = 2; j <= n; j++){
cur[j] += cur[j-1] ;
}
}
return cur[n];
}
}
提交结果: