牛客地址:牛妹的礼物
一、思路:
用一个矩阵dp来保存走到每个格子的时候,当前格子累计的礼物的最小体积,dp的大小和格子的大小一致,也是N*M的矩阵。
二、图示:
(图片来自牛客大佬)
三、详细流程:
1、dp第0行和第0列的初始化;
2、dp的更新,注意题目中要求只能想右、下、右下角走。所以dp[i][j]是它的左上角、上、左,三个值中最小的一个,再加上当前格子的礼物体积:
presentVolumn[i][j] += Math.min(presentVolumn[i-1][j-1], Math.min(presentVolumn[i-1][j], presentVolumn[i][j-1]));
3、返回值是dp最右下角的值。
四、代码:
import java.util.*;
public class Solution {
/**
*
* @param presentVolumn int整型二维数组 N*M的矩阵,每个元素是这个地板砖上的礼物体积
* @return int整型
*/
public int selectPresent (int[][] presentVolumn) {
// write code here
if(presentVolumn.length==0 || presentVolumn[0].length==0 || presentVolumn==null){
return 0;
}
int row = presentVolumn.length;
int col = presentVolumn[0].length;
// 动态规划:dp[i][j]:到达presentVolumn[i][j]时,最小的体积和
//初始化第0行
for (int i = 1; i < col; i++) {
presentVolumn[0][i] += presentVolumn[0][i-1];
}
//初始化第0列
for (int i = 1; i < row; i++) {
presentVolumn[i][0] += presentVolumn[i-1][0];
}
//计算内部值
for (int i = 1; i < row; i++) {
for (int j = 1; j < col; j++) {
// 走到第(i,j)的最小体积dp[i][j] = min{左边一格的最小体积,上边一格的最小体积,左上一格的最小体积}
presentVolumn[i][j] += Math.min(presentVolumn[i-1][j-1], Math.min(presentVolumn[i-1][j], presentVolumn[i][j-1]));
}
}
return presentVolumn[row-1][col-1];
}
}
这个题目有点小瑕疵,明明说了 N>0,然而测试用例还出现N为0的情况