leetcode-152. 乘积最大子数组

题目

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

示例 1:

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

示例 2:

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

解题思路

看官方题解后才会做的题。

重点：如果维护一个保存连乘乘积最大的数组，则当前位置的最优解未必是由前一个位置的最优解转移得到的
比如[2,-3,2,-3]的最大乘积应该是全部都乘，如果只看前一位，则max数组为[2,-3,2,-3]，最大变成2了。

对于乘法而言，因为乘积会随着符号改变，所以同时维护2个数组，1个是当前位置的最大乘积，1个是当前位置的最小乘积，新增的最大乘积，同时考虑前面的最小、最大即可。

代码

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        product_max = [1]
        product_min = [1]
        for index, each_num in enumerate(nums):
            last_max, last_min = product_max[-1], product_min[-1]
            product_max.append(max(each_num, each_num * last_max, each_num * last_min))
            product_min.append(min(each_num, each_num * last_max, each_num * last_min))
        print(product_max)
        print(product_min)
        return max(product_max[1:])

因为当前状态只和前一个状态有关，所以可以缩减空间，变为：

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        last_max, last_min = 1, 1
        ret_max = nums[0]
        for index, each_num in enumerate(nums):
            last_max, last_min = max(each_num, each_num * last_max, each_num * last_min), min(each_num, each_num * last_max, each_num * last_min)
            ret_max = max(ret_max, last_max)
        return ret_max