题目要求:
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积 。
示例1:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例2:
输入: [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
题解
思路1:
对于整个数组,我们可以构建一个二维记事本,用于存储所有从 i 到 j 可能的序列的乘积,最终找出最大值即可。
class Solution:
def maxProduct(self, nums: List[int]) -> int:
l = len(nums)
max = nums[0]
product = [[0 for _ in range(l)] for _ in range(l)]
for i in range(l):
product[i][i] = nums[i]
if product[i][i] > max:
max = product[i][i]
for i in range(l-1):
for j in range(i+1,l):
product[i][j] = nums[j] * product[i][j-1]
if product[i][j] > max:
max = product[i][j]
return max
但是这样做对于内存有一定要求,且当输入数据长度很长的时候,容易打不到题目设计的实验要求。
思路2:
维护三个值,一个为当前位置之前的乘积最大值maxj,一个为当前位置之前的乘积最小值mini,一个为全局的最大值maxt。 对于maxj,如果当前值为负,则我们希望和之前乘积的最小值mini(最可能为负)相乘,以获得全局最大值。如果当前值为正,则希望与之前的乘积最大值maxj(最可能为正)相乘,以获得全局最大值,并且乘积获得的值要和当前值进行比较,选择更大的值作为下一位置的maxj。mini更新也同理。且maxj和maxt作比较更新maxt。
class Solution:
def maxProduct(self, nums: List[int]) -> int:
l = len(nums)
mini,maxj, maxt = 1,1,nums[0]
for i in range(l):
if nums[i]<0:
mini,maxj = maxj,mini
maxj = max(nums[i]*maxj,nums[i])
mini = min(nums[i]*mini,nums[i])
maxt = max(maxj,maxt)
return maxt