题目:
剑指 Offer 47. 礼物的最大价值。
在一个 m*n 的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于 0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值,请计算你最多能拿到多少价值的礼物?
链接:https://leetcode-cn.com/leetbook/read/illustrate-lcof/xstkx3/
示例:
输入:
[
[1,3,1],
[1,5,1],
[4,2,1]
]
输出: 12
解释: 路径 1→3→5→2→1 可以拿到最多价值的礼物
提示:
0 < grid.length <= 200
0 < grid[0].length <= 200
思路:
重新为数组赋值,最后返回数组右下角的元素值。
遍历数组中的元素,除右上角元素外,每个元素更新为其与左侧或上方和的最大值。第一行和第一列作特殊处理,第一行不需要考虑上方,第一列不需要考虑左侧。
代码:
class Solution:
def maxValue(self, grid: List[List[int]]) -> int:
m, n = len(grid), len(grid[0])
for i in range(m):
for j in range(n):
if i == 0 and j == 0:
continue
elif i == 0 and j != 0:
grid[i][j] += grid[i][j-1]
elif i != 0 and j==0:
grid[i][j] += grid[i-1][j]
else:
grid[i][j] += max(grid[i-1][j], grid[i][j-1])
return grid[m-1][n-1]
时间复杂度O(n),空间复杂度O(1).