Leetcode152. 乘积最大子数组——python

题目

给你一个整数数组 nums ，请你找出数组中乘积最大的连续子数组（该子数组中至少包含一个数字），并返回该子数组所对应的乘积。

示例

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

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

来源

链接：https://leetcode-cn.com/problems/maximum-product-subarray

代码

1、暴力求解

class Solution:
    def maxProduct(self, A):
        B = A[::-1]
        for i in range(1, len(A)):
            A[i] *= A[i - 1] or 1
            B[i] *= B[i - 1] or 1
        return max(max(A),max(B)) 

2、动态规划

class Solution(object):
    def maxProduct(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        maxF,minF,ans = nums[0],nums[0],nums[0]
        n = len(nums)
        for i in range(1,n):
            mx,mn = maxF,minF
            maxF = max(mx*nums[i],max(nums[i],mn*nums[i]))
            minF = min(mn*nums[i],min(nums[i],mx*nums[i]))
            ans = max(maxF,ans)
        return ans

3、考虑动态规划的优化

class Solution:
    def maxProduct(self,nums):
        imax,imin = 1,1
        n = len(nums)
        ans = float('-Inf')
        for i in range(n):
            if nums[i]<0:
                imax,imin = imin,imax
            imax = max(imax * nums[i],nums[i])
            imin = min(imin * nums[i],nums[i])
            ans = max(ans,imax)
        return ans