【面试题47 礼物的最大价值】
在一个 m*n 的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于 0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值,请计算你最多能拿到多少价值的礼物?
Leetcode题目对应位置: 面试题47:礼物的最大价值
思路:动态规划问题。
dp[i][j]
表示从左上角走到当前位置 (i,j)
能获取到的最大价值dp[i][j] = max(dp[i-1][j], dp[i][j-1]) + grid[i][j]
时间复杂度:O(mn),遍历整个数组
空间复杂度:O(mn),需要创建与原数组等大的 dp 数组
class Solution:
def maxValue(self, grid: List[List[int]]) -> int:
if not grid: return 0
r, c = len(grid), len(grid[0])
# dp数组初始化
dp = [[0] * c for _ in range(r)]
dp[0][0] = grid[0][0]
for i in range(1, r):
dp[i][0] = dp[i-1][0] + grid[i][0]
for j in range(1, c):
dp[0][j] = dp[0][j-1] + grid[0][j]
# 移动到右下角
for i in range(1, r):
for j in range(1, c):
dp[i][j] = max(dp[i-1][j], dp[i][j-1]) + grid[i][j]
return dp[-1][-1]
空间复杂度优化:可以发现 dp[i][j]
仅和 dp[i-1][j]
和 dp[i][j-1]
有关,可以直接在原矩阵 grid 上进行修改,可以将空间复杂度降到 O(1)。
class Solution:
def maxValue(self, grid: List[List[int]]) -> int:
if not grid: return 0
r, c = len(grid), len(grid[0])
# 初始化
for i in range(1, r):
grid[i][0] += grid[i - 1][0]
for j in range(1, c):
grid[0][j] += grid[0][j - 1]
# dpr
for i in range(1, r):
for j in range(1, c):
grid[i][j] += max(grid[i][j - 1], grid[i - 1][j])
return grid[-1][-1]