题目要求
在一个 m*n 的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于 0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值,请计算你最多能拿到多少价值的礼物?
解题思路
动态规划
d p [ i ] [ j ] dp[i][j] dp[i][j]代表走到第i行第j列礼物的最大价值。
class Solution(object):
def maxValue(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
# 创建数组
dp = []
# 初始化数组
for i in range(len(grid)):
dp.append([])
for j in range(len(grid[0])):
dp[i].append(0)
# 求dp数组的第一行和第一列
dp[0][0] = grid[0][0]
for j in range(1, len(grid[0])):
dp[0][j] = dp[0][j - 1] + grid[0][j]
for i in range(1, len(grid)):
dp[i][0] = dp[i - 1][0] + grid[i][0]
# 求数组其他位置元素
for i in range(1, len(grid)):
for j in range(1, len(grid[0])):
dp[i][j] = max(dp[i - 1][j], dp[i][j - 1]) + grid[i][j]
# 返回最终值
return dp[len(grid) - 1][len(grid[0]) - 1]
经观察发现,数组只与上面元素和左边元素有关,因此可以直接在原数组上修改,无需创建新的数组dp:
动态规划(改进空间复杂度)
class Solution(object):
def maxValue(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
# 计算第一行和第一列元素
for j in range(1, len(grid[0])):
grid[0][j] += grid[0][j-1]
for i in range(1, len(grid)):
grid[i][0] += grid[i-1][0]
# 计算其他行列的元素
for i in range(1, len(grid)):
for j in range(1, len(grid[0])):
grid[i][j] += max(grid[i-1][j], grid[i][j-1])
# 返回要求值
return grid[len(grid)-1][len(grid[0])-1]