【LeetCode152】乘机最大子数组

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

 

示例 1:

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

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

 

先定义一个数组dpMax，用dpMax[i] 表示以第i个元素的结尾的子数组， 乘机最大的值，也就是这个数组必须包含i个元素。

那么dpMax[i] 的话有几种取值：

当nums[i] >= 0 并且 dpMax[i-1] > 0 ，dpMax[i] = dpMax[i] * nums[i]

当nums[i] >= 0 并且dpMax[i-1] < 0， 此时如果和前边的数累乘的话，会变成负数，所以dpMax[i] = nums[i]

当nums[i] < 0, 此时如果前边累乘结果是一个很大的负数， 和当前负数累乘的话就会变成一个更大的数。所以我们还需要一个数组dpMin来记录以第i个元素的结尾的子数组，乘积最小的值。

               当dpMin[i-1] < 0， dpMax[i] = dpMin[i-1] * nums[i]

               当dpMin[i-1] >= 0, dpMax[i] = nums[i]

当然，上边引入了dpMin数组，怎么求dpMin其实和上边求dpMax的过程其实是一样的。

按上边的分析，我们就需要加很多的if else来判断不同的情况，这里可以用个技巧。

我们注意到上边dpMax[i] 的取值无非就是三种，dpMax[i-1] * nums[i] 、dpMin[i-1] * nums[i] 以及 nums[i]。

我们更新的时候，无需去区分当前是哪种情况，只需要从三个值中选一个最大的即可。

dpMax[i] = max(dpMax[i-1] * nums[i], dpMin[i-1] * nums[i], nums[i]);

求dpMin[i]同理。

dpMin[i] = min(dpMax[i-1] * nums[i], dpMin[i-1] * nums[i], nums[i]);

更新过程中，我们可以用一个变量去保存当前得到的最大值，因为它只依赖它前面一个值

func maxProduct(nums []int) int {
    maxF, minF, ans := nums[0], nums[0], nums[0]
    for i := 1; i < len(nums); i++ {
        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
}

func max(x, y int) int {
    if x > y {
        return x
    }
    return y
}

func min(x, y int) int {
    if x < y {
        return x
    }
    return y
}