题目
在一个 m*n
的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于 0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值,请计算你最多能拿到多少价值的礼物?
示例 1:
输入:
[
[1,3,1],
[1,5,1],
[4,2,1]
]
输出: 12
解释: 路径 1→3→5→2→1 可以拿到最多价值的礼物
思路
典型的动态规划
- 定义dp数组
dp[i][j]
代表当前走到格子的(i - 1, j - 1)
位置时,能拿到的最多价值的礼物
- 确定状态转移方程
dp[i][j] = Max(dp[i - 1][j], dp[i][j - 1]) + value(i - 1, j - 1)
- 每次可以向右或者向下移动,所以在走到
(i - 1, j - 1)
位置时,需要考虑相邻的上边以及左边位置所能拿到的最大值,取两个中最大的,再加上当前位置的值,即为走到当前位置时所能拿的最大的值
- 初始化
dp[1][1] = grid[0][0]
,dp[0][]
以及dp[][0]
时,初始化为0
class Solution {
public int maxValue(int[][] grid) {
if (grid == null || grid.length <= 0) return 0;
int[][] dp = new int[grid.length + 1][grid[0].length + 1];
for (int i = 1; i <= grid.length; i++) {
for (int j = 1; j <= grid[i].length; j++) {
dp[i][j] = Math.max(dp[i - 1][j], dp[i][j - 1]) + grid[i - 1][j - 1];
}
}
return dp[grid.length][grid[0].length];
}
}
复杂度分析
- 时间复杂度O(n * m)
- 空间复杂度O((n+1) * (m+1)),另外开辟了一个二维数组
进一步优化(对二维数组进行降维)
由于dp[i][j]
只与dp[i - 1][j]、dp[i][j - 1]
有关,所以第i - 2
行以及其上面的所有行都没必要存储,所以可以将二维数组优化为一维数组,需要注意的是,一维数组是滚动变化的
class Solution {
public int maxValue(int[][] grid) {
if (grid == null || grid.length <= 0) return 0;
int[] dp = new int[grid[0].length + 1];
for (int i = 1; i <= grid.length; i++) {
for (int j = 1; j <= grid[0].length; j++) {
dp[j] = Math.max(dp[j - 1], dp[j]) + grid[i - 1][j - 1];
}
}
return dp[grid[0].length];
}
}
复杂度分析
- 时间复杂度O(m * n)
- 空间复杂度O(n+1),另外开辟了一个一维数组,数组长度为n+1