Leetcode 152. 乘积最大子数组【动态规划】

问题描述

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

示例 1:

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

解题报告

这道题需要注意的点有两个：

• 在遇到一个负数时，当前累计乘积的最小值就会变成当前乘积的最大值，而当前累计乘积的最大值会变成最小值，所以本题不仅需要记录当前累计乘积的最大值，还要记录当前累计乘积的最小值。
• 求当前乘积的最大值时，什么时候开始新的起点呢？当当前乘积的最大值与当前元素的乘积比当前元素还要小时，就要开始新的起点了【这一点和Leetcode 53. 最大子序和非常相似】。求当前累计乘积的最小值同样的操作。

实现代码

class Solution {
    public:
        int maxProduct(vector<int>&nums) {
            int ans= INT_MIN, max_= 1, min_ = 1;
            for(int i=0; i<nums.size(); i++){
                // 这一步是为了维持当前累计乘积的最大值和最小值
                if(nums[i] < 0) swap(max_, min_);
                // 这一步是决定是否重新新的起点
                max_= max(max_*nums[i], nums[i]);
                min_= min(min_*nums[i], nums[i]);
                
                ans= max(ans, max_);
            }
            return ans;
    }
};

参考资料

[1] Leetcode 152. 乘积最大子数组