152.maximum-product-subarray


152. 乘积最大子序列

题目地址

给定一个整数数组 nums ,找出一个序列中乘积最大的连续子序列(该序列至少包含一个数)。

示例 1:

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

示例 2:

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

思路

DP 方程需要两个维度

// DP[i, 0], DP[i, 1] , 0 表示 max, 1 表示 min
    if a[i] >= 0:
        DP[i, 0] = DP[i - 1, 0] * a[i]
        DP[i, 1] = DP[i - 1, 1] * a[i]
    else:
        DP[i, 0] = DP[i - 1, 1] * a[i]
        DP[i, 1] = DP[i - 1, 0] * a[i]

优化:

要保存最大值, 最小值。因为负负得正

解法一

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        if nums is None: return 0

        dp = [[0 for _ in range(2)] for _ in range(2)]

        dp[0][1], dp[0][0], res = nums[0], nums[0], nums[0]

        for i in range(1, len(nums)):
            x, y = i % 2, (i - 1) % 2 # 滚动数组
            dp[x][0] = max(dp[y][0] * nums[i], dp[y][1] * nums[i], nums[i])
            dp[x][1] = min(dp[y][0] * nums[i], dp[y][1] * nums[i], nums[i])
            res = max(res, dp[x][0])

        return res
class Solution {
    public int maxProduct(int[] nums) {
        int min = nums[0], max = nums[0], result = nums[0];
        for (int i = 1; i < nums.length; i++) {
            if(nums[i] < 0) {
                int tmp = 0;
                tmp = min;
                min = max;
                max = tmp;
            }
            min = Math.min(min * nums[i], nums[i]);
            max = Math.max(max * nums[i], nums[i]);
            result = Math.max(result, max);
        }
        return result;
    }
}
class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        mi = ma = res = nums[0]
        for i in range(1, len(nums)):
            if nums[i] < 0: mi, ma = ma, mi
            ma = max(ma * nums[i], nums[i])
            mi = min(mi * nums[i], nums[i])
            res = max(res, ma)

        return res
class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        if nums is None: return 0

        res, curMax, curMin = nums[0], nums[0], nums[0]

        for i in range(1, len(nums)):
            num = nums[i]
            curMax, curMin = curMax * num, curMin * num
            curMin, curMax = min(curMax, curMin, num), max(curMax, curMin, num)
            res = curMax if curMax > res else res
        return res