Leetcode152.乘积最大子数组

152. 乘积最大子数组

给你一个整数数组 nums ,请你找出数组中乘积最大的非空连续

子数组

(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。

测试用例的答案是一个 32-位 整数。

示例 1:

1
2
3
>输入: nums = [2,3,-2,4]
>输出: 6
>解释: 子数组 [2,3] 有最大乘积 6。

示例 2:

1
2
3
>输入: nums = [-2,0,-1]
>输出: 0
>解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。

令imax为当前最大值,则当前最大值为 imax = max(imax * nums[i], nums[i])

由于存在负数,那么会导致最大的变最小的,最小的变最大的。因此还需要维护当前最小值imin,imin = min(imin * nums[i], nums[i])

当负数出现时则imax与imin进行交换再进行下一步计算

代码:

1
2
3
4
5
6
7
8
9
10
11
12
class Solution(object):
def maxProduct(self, nums):
max_ans = 1
min_ans = 1
res = -10**4
for num in nums:
if num < 0:
max_ans, min_ans = min_ans, max_ans
max_ans = max(max_ans * num, num)
min_ans = min(min_ans * num, num)
res = max(max_ans, res)
return res