题目描述:
在一个m×n的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格直到到达棋盘的右下角。给定一个棋盘及其上面的礼物,请计算你最多能拿到多少价值的礼物?
分析:
与最小路径规划那个题类似,这是最大路径规划的问题。求解的方法借助于一个dp数组,对首元素赋值,对第一行和第一列赋值,剩下的元素取决于两个方向上其前面元素的最大值。注意:最小路径规划类似的方法,局限于移动方向固定两个的场景。
public static int maxPathSum(int [][] a){
int N=a.length;
int M=a[0].length;
int[][] dp=new int[N][M];
dp[0][0]=a[0][0];
for(int i=1;i<M;i++){
dp[0][i]=a[0][i]+dp[0][i-1];
}
for(int i=1;i<N;i++){
dp[i][0]=a[i][0]+dp[i-1][0];
}
for(int i=1;i<a.length;i++){
for(int j=1;j<a[i].length;j++){
dp[i][j]=Math.max(dp[i-1][j],dp[i][j-1])+a[i][j];
}
}
return dp[N-1][M-1];
}
优化:
优化空间复杂度,使用一维数组
题目中可知,坐标(i,j)的最大礼物价值仅仅取决于坐标为(i-1,j)和(i,j-1)两个格子;
因此第i-2行以上的最大价值没有必要保存下来。
使用一维数组保存:(0,0)…(0,j-1)保存的是(i,0)…(i,j-1)的最大价值;(0,j)…(0,cols-1)保存的是(i-1,j)…(i-1,cols-1)的最大价值。
public static int maxPathSum(int [][] a){
int rows=a.length;//行数
int cols=a[0].length;//列数
int[] dp=new int[cols]; //建立列数大小的dp一维数组
for(int i=0;i<rows;i++) {
for(int j=0;j<cols;j++) {
int up = 0;//当行数i=0时,上边up=0
int left = 0;//当列数j=0时,左边left=0
if(i>0)
up = dp[j];
if(j>0)
left = dp[j-1];
maxValue[j] = Math.max(up, left)+arr[i][j];
}
}
return dp[cols-1];
}