给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
示例 1:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-product-subarray
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
————————————
解题思路:使用动态规划,维护两个状态变量max_prev和min_prev,max_prev表示以当前节点结尾的连续子数组的最大值,min_prev表示以当前节点结尾的连续子数组的最小值。
对于数组中的下一个数字,以其结尾的最大连续子数组可能是其本身,也可能是num[i]*max_prev,也可能是num[i]*min_prev。因此我们可以得到状态转移方程: f m a x ( i ) = m a x ( f m a x ( i − 1 ) ∗ n u m [ i ] , f m i n ( i − 1 ) ∗ n u m [ i ] , n u m [ i ] ) f_{max}(i)=max(f_{max}(i-1)*num[i],f_{min}(i-1)*num[i],num[i]) fmax(i)=max(fmax(i−1)∗num[i],fmin(i−1)∗num[i],num[i]) f m i n ( i ) = m i n ( f m a x ( i − 1 ) ∗ n u m [ i ] , f m i n ( i − 1 ) ∗ n u m [ i ] , n u m [ i ] ) f_{min}(i)=min(f_{max}(i-1)*num[i],f_{min}(i-1)*num[i],num[i]) fmin(i)=min(fmax(i−1)∗num[i],fmin(i−1)∗num[i],num[i])
基于以上状态转移方程有:
# Python3
class Solution:
def maxProduct(self, nums: List[int]) -> int:
if not nums:
return None
max_prev = nums[0] # 初始化连续子数组最大值
min_prev = nums[0] # 初始化连续子数组最小值
length = len(nums)
max_ans = nums[0] # 初始化返回值
for i in range(1,length):
max_temp= max_prev
min_temp = min_prev
max_prev = max(max_temp*nums[i],min_temp*nums[i],nums[i]) # 状态转移方程
min_prev = min(max_temp*nums[i],min_temp*nums[i],nums[i])
max_ans = max(max_ans,max_prev) # 更新当前连续子数组的最大值
return max_ans