剑指 Offer 42. 连续子数组的最大和
输入一个整型数组,数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为O(n)。
示例1:
输入: nums = [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
提示:
1 <= arr.length <= 10^5
-100 <= arr[i] <= 100
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
res, cur = nums[0], nums[0]
for val in nums[1:]:
cur = max(cur+val, val) # 关键的步骤:cur 要么等于之前之和,要么从当前开始
res = max(res, cur)
return res
剑指 Offer 47. 礼物的最大价值
在一个 m*n 的棋盘的每一格都放有一个礼物,每个礼物都有一定的价值(价值大于 0)。你可以从棋盘的左上角开始拿格子里的礼物,并每次向右或者向下移动一格、直到到达棋盘的右下角。给定一个棋盘及其上面的礼物的价值,请计算你最多能拿到多少价值的礼物?
示例 1:
输入:
[
[1,3,1],
[1,5,1],
[4,2,1]
]
输出: 12
解释: 路径 1→3→5→2→1 可以拿到最多价值的礼物
提示:
0 < grid.length <= 200
0 < grid[0].length <= 200
动态规划解析:
-
状态定义: 设动态规划矩阵 dp ,dp(i,j)代表从棋盘的左上角开始,到达单元格 (i,j) 时能拿到礼物的最大累计价值。
-
转移方程:
-
当 i = 0 且 j = 0 时,为起始元素;
-
当 i = 0 且 j !=0 时,为矩阵第一行元素,只可从左边到达;
-
当 i != 0 且 j = 0 时,为矩阵第一列元素,只可从上边到达;
-
当 i !=0 且 j !=0 时,可从左边或上边到达;
-
-
初始状态: dp[0][0] = grid[0][0] ,即到达单元格 (0,0) 时能拿到礼物的最大累计价值为 grid[0][0] ;
-
返回值: dp[m-1][n−1] ,m, n 分别为矩阵的行高和列宽,即返回 dp 矩阵右下角元素。
class Solution:
def maxValue(self, grid: List[List[int]]) -> int:
for i in range(len(grid)):
for j in range(len(grid[0])):
if i == 0 and j == 0: continue
if i == 0: grid[i][j] += grid[i][j - 1]
elif j == 0: grid[i][j] += grid[i - 1][j]
else: grid[i][j] += max(grid[i][j - 1], grid[i - 1][j])
return grid[-1][-1]
# 优化
class Solution:
def maxValue(self, grid: List[List[int]]) -> int:
m, n = len(grid), len(grid[0])
for j in range(1, n): # 初始化第一行
grid[0][j] += grid[0][j - 1]
for i in range(1, m): # 初始化第一列
grid[i][0] += grid[i - 1][0]
for i in range(1, m):
for j in range(1, n):
grid[i][j] += max(grid[i][j - 1], grid[i - 1][j])
return grid[-1][-1]