在一个m×n的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于 0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值,请计算你最多能拿到多少价值的礼物?
v1
从题目说的思路,我们可以列出这样一个方程:
maxReward(n,n) = max(maxReward[n,n+1]+M[n,n+1]) + max(maxReward[n+1,n]+M[n+1,n])
终止条件maxReward(M[1,1]) = M[1,1]
这样的思路相当于把一个很复杂的问题,换成一个个可以求解的小问题。
但是这样做显然复杂度太高了,到O(2^n)。因为有很多maxReward会重复计算。
v2 – DP
换成一种比较容易想到的思路,反过来计算maxReward(1,1),然后对于maxReward(2,1),maxReward(1,2)都是很好求解的。之后当maxReward(2,2)出现时,那只有可能是来自maxReward(2,1),或者maxReward(1,2)之间较大的一个。
以此类推,我们按照上面所提到的转移方程,通过循环的方式实现。
class Solution:
def maxValue(self , grid: List[List[int]]) -> int:
n, m = len(grid), len(grid[0])
maxReward = [[0]*m]*n
for i in range(n-1, -1, -1):
for j in range(m-1, -1, -1):
maxVal = 0
if i+1<n: #防止越界访问
maxVal = maxReward[i+1][j]
if j+1<m:
maxVal = max(maxVal, maxReward[i][j+1])
maxReward[i][j] = maxVal + grid[i][j]
return maxReward[0][0]
评价
这个解法比最高频出现的解法,先把最后一行和最后一列算一遍的方法,要更简洁一些。
是因为在循环中增加了越界访问的判断,并且顺序始终是从一行一行完成的。这样可以保证求解是绝对正确的。