面试题47. 礼物的最大价值 python3 动态规划 递推公式

题目

https://leetcode-cn.com/problems/li-wu-de-zui-da-jie-zhi-lcof/

在一个 m*n 的棋盘的每一格都放有一个礼物，每个礼物都有一定的价值（价值大于 0）。你可以从棋盘的左上角开始拿格子里的礼物，并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值，请计算你最多能拿到多少价值的礼物？

示例 1:

输入: 
[
  [1,3,1],
  [1,5,1],
  [4,2,1]
]
输出: 12
解释: 路径 1→3→5→2→1 可以拿到最多价值的礼物

提示：

0 < grid.length <= 200
0 < grid[0].length <= 200

思路

每一个位置[i,j]的最大值res[i][j]基本都依赖于它之前位置的最大值，即res[i-1][j]和res[i][j-1]。

因此，还是和前面的把数字翻译成字符串、上楼梯、斐波那契数列等题一样，用动态规划+递推公式求解。

递推关系为：

边界值和初始值：

当i=0, j=0时，res的初始值即为输入数组grid[0][0]的值；

当i=0, j!=0时，res[i][j] = res[i][j-1] + grid[i][j], 理由是此时的路线只能是从左到右而来，因此只与左边的值有关；

同理，当i!=0, j=0时，res[i][j] = res[i-1][j] + grid[i][j];

内部：

对于i>0, j>0, 通过观察可以发现，res[i][j] = max(res[i-1][j] ,res[i][j-1] ) + grid[i][j]

 

代码

由上述的思路，即可完成代码

class Solution:
    def maxValue(self, grid: List[List[int]]) -> int:
        m = len(grid)
        n = len(grid[0])
        res = [[0]*n for i in range(m)] # 创建最大价值矩阵
        # 初始化[0,0]
        res[0][0] = grid[0][0]
        # 初始化[0,X]
        for j in range(1,n):
            res[0][j] = res[0][j-1] + grid[0][j]
        # 初始化[X,0]
        for i in range(1,m):
            res[i][0] = res[i-1][0] + grid[i][0]
        # 动态规划
        for i in range(1,m):
            for j in range(1,n):
                res[i][j] = max(res[i][j-1],res[i-1][j]) + grid[i][j]

        return res[-1][-1]

复杂度

时间复杂度O(MN)

空间复杂度O(MN)

如果要节约空间，可以将grid原地转换为res.

结果