LeetCode-152. 乘积最大子数组

动态规划#

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

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

请注意，一个只包含一个元素的数组的乘积是这个元素的值。

示例 1:

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

示例 2:

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

思路

这题其实只需要解决掉负数这个问题即可。

如果同时处理curMax和curMin和nums[i]的正负性比较复杂，我们只需要利用max和min比较即可

1
func maxProduct(nums []int) int {
2
    if len(nums) == 0 {
3
        return 0
4
    }
5

6
    curMax := nums[0]
7
    curMin := nums[0]
8
    res := nums[0]
9

10
    for i := 1; i < len(nums); i++ {
11
        tmpMax, tmpMin := curMax, curMin
12

13
        curMax = max(nums[i], max(tmpMax*nums[i], tmpMin*nums[i]))
14
        curMin = min(nums[i], min(tmpMax*nums[i], tmpMin*nums[i]))
15

16
        res = max(res, curMax)
17
    }
18
    return res
19
}