152. 乘积最大数组
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
示例
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
输入: [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
解题思路
看完这个解法觉得woc这题也能算medium?最最最关键的一句话:由于存在负数,那么会导致最大的变最小的,最小的变最大的。因此还需要维护当前最小值imin。
正是这个原因,当即将要乘的这个数字是负数,先要把已经记录的最大值与最小值调换——最小值乘完一个负数反而成了最大。
递推公式也很简单,“乘还是不乘,这是个问题”:
imax = max(imax * num, num)
imin = max(imin * num, num)(完全是为了找最大值服务)
最后,真正的最大值为Max = max(Max, imax)
原解答用的是Java,这里依旧改成Python:
class Solution:
def maxProduct(self, nums: List[int]) -> int:
Max, imax, imin = -float('inf'), 1, 1
for num in nums:
if num < 0:
temp = imax
imax = imin
imin = temp
imax = max(imax * num, num)
imin = min(imin * num, num)
Max = max(Max, imax)
return Max
遍历了一遍数组,时间复杂度是O(n);
用了额外四个变量存值,空间复杂度是O(1).
打完收工!