题目描述:
给你一个整数数组 nums
,请你找出数组中乘积最大的非空连续 子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
测试用例的答案是一个 32-位 整数。
示例 1:
输入: nums = [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例 2:
输入: nums = [-2,0,-1] 输出: 0 解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
提示:
1 <= nums.length <= 2 * 104
-10 <= nums[i] <= 10
nums
的任何子数组的乘积都 保证 是一个 32-位 整数
我的作答:
踩坑总结:(1)要保证连续,动态数组元素必须有nums[i]参与,如果储存的是nums[i]之前的最大乘积是不行的!!!(2)容易陷入局部最优,如果只是单纯的max(dp[i-1]*nums[i], nums[i])面对[-1 1 -4]这样的数组就不能通过,因为万一再碰到负数就容易翻盘!
所以使用两个动态变量,一个储存最大值,一个储存最小值,保证“负负得正”翻盘的机会!
class Solution:
def maxProduct(self, nums: List[int]) -> int:
if not nums: return 0
res = nums[0] #保存最后的结果
pre_max = nums[0]
pre_min = nums[0]
for i in range(1, len(nums)):
cur_max = max(pre_max*nums[i], pre_min*nums[i], nums[i])
cur_min = min(pre_min*nums[i], pre_max*nums[i], nums[i])
pre_max = cur_max
pre_min = cur_min
if pre_max>res:
res = pre_max
return res
参考:
无