152. 乘积最大子数组
1.题目描述
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字)。
示例 1:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例 2:
输入: [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-product-subarray
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.思路
看到题目的时候,以为和最大子数组之和一样,开辟dp数组,采用动态规划策略直接就能作。看到题目难度为中等,我就知道事情没有那么简单。
看了高赞官方题解,其维护了max, imax, imin
变量,分别表示当前最终最大乘积、当前最大乘积值,当前最小乘积值。由于
n
u
m
s
[
i
]
nums[i]
nums[i]可能为负,而乘积受负数影响,两个数之间的大小关系会变化。针对这个问题,采用同时计算,维护当前状态的最大、最小值。
至于为什么当
n
u
m
s
[
i
]
<
0
nums[i]<0
nums[i]<0时,imax,imin的值要交换的原因,我想是:
当
a
>
b
,
c
<
0
a>b,c<0
a>b,c<0时,有
a
∗
c
<
b
∗
c
a*c<b*c
a∗c<b∗c
3.代码
def maxProduct(self, nums: List[int]) -> int:
if len(nums) == 0:
return 0
# 初始化辅助变量
imax = 1
imin = 1
max_val = float("-inf") # 最后的连续子数组的最大成绩值
for i in range(len(nums)):
if (nums[i]<0):
imax, imin = imin, imax # 交换最大、最小值
imax = max(imax*nums[i], nums[i])
imin = min(imin*nums[i], nums[i])
max_val = max(max_val, imax)
return max_val