【面试题47 礼物的最大价值】
在一个 m*n 的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于 0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值,请计算你最多能拿到多少价值的礼物?
Leetcode题目对应位置: 面试题47:礼物的最大价值
思路:动态规划问题。
- 明确状态:棋盘上每个格子的状态是走到当前格子能拿到的最大价值
- dp 数组:是一个和棋盘大小相同的矩阵,每个位置
dp[i][j]
表示从左上角走到当前位置(i,j)
能获取到的最大价值 - 明确选择:每个位置只能向下或向右走,那么同样地,每个位置都是由上面或左侧的位置移动而来
- 状态间的关系:知道了选择,就可知当前位置最大价值 = 上面和左侧的较大者 + 当前位置的价值,即
dp[i][j] = max(dp[i-1][j], dp[i][j-1]) + grid[i][j]
- 确定 base case:在左上角位置,还未开始移动,价值就是本身;在第一行只能由其左侧移动而来,在第一列只能由其上侧移动而来,所以第一行/第一列位置的值就是其左侧/上侧的累加和。
时间复杂度: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]