152. 乘积最大子数组(python3)

题目:https://leetcode-cn.com/problems/maximum-product-subarray/

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

 

示例 1:

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

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

 

解题思路:

这道题乍看上去比较像https://blog.csdn.net/anningne/article/details/111680520这道题,但写起来会发现大错特错,因为都是整数,所以影响大小的重点在于负数,菜鸟的我实在太菜,参考了https://leetcode-cn.com/problems/maximum-product-subarray/solution/hua-jie-suan-fa-152-cheng-ji-zui-da-zi-xu-lie-by-g/的解法,核心在于当出现负数时,最小变为最大,最大变为最小。在存储最大值maxvalue时需要在循环内判断,而不是像https://blog.csdn.net/anningne/article/details/111680520最后循环遍历,因为dp_max[i-1]可能会在下一循环被dp_min[i-1]冲掉

 

代码:

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        m = len(nums)
        if(m == 0):
            return 0
        dp_max = [0] * m
        dp_min = [0] * m

        dp_max[0] = nums[0]
        dp_min[0] = nums[0]
        maxvalue = dp_max[0]
 
        for i in range(1, m):
            if(nums[i] < 0):
                temp = dp_max[i - 1]
                dp_max[i - 1] = dp_min[i - 1]
                dp_min[i - 1] = temp
            
            dp_max[i] = max(dp_max[i - 1] * nums[i], nums[i])
            dp_min[i] = min(dp_min[i - 1] * nums[i], nums[i])
            maxvalue = dp_max[i] if maxvalue < dp_max[i] else maxvalue
    

        return maxvalue

 

测试:

if __name__ == "__main__":
    nums =[-3,-1,-1]
    res = Solution().maxProduct(nums)
    print(res)

 

你可能感兴趣的:(DP,leetcode,medium,leetcode)