老卫带你学---leetcode刷题(152. 乘积最大子数组)

152. 乘积最大子数组

问题:

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

测试用例的答案是一个 32-位 整数。

子数组 是数组的连续子序列。

示例 1:

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

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

1 <= nums.length <= 2 * 104
-10 <= nums[i] <= 10
nums 的任何前缀或后缀的乘积都 保证 是一个 32-位 整数

解决:

dp

我们只要记录前 i 的最小值,和最大值,那么 dp[i] = max(nums[i] * pre_max, nums[i] * pre_min, nums[i]),这里 0 不需要单独考虑,因为当相乘不管最大值和最小值,都会置 0

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        res,pre_max,pre_min = nums[0],nums[0],nums[0]
        for i in nums[1:]:
            cur_max=max(pre_max*i,pre_min*i,i)
            cur_min=min(pre_max*i,pre_min*i,i)
            res = max(res,cur_max)
            pre_max=cur_max
            pre_min=cur_min
        return res

你可能感兴趣的:(leetcode)