152. 乘积最大子数组

· 约 1 分钟阅读 · 次阅读 leetcode

给你一个整数数组 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_max[i]是以第i个数结尾的乘积最大子数组的积,则 dp_max[i] = max(dp_max[i-1]*nums[i], nums[i])

但是这样只能解决数组中所有数全部非负或全部非正的情形。

考虑负负得正的情形,我们新加一个dp_min, 表示最小的积,

则:

dp_max[i] = max(dp_max[i-1]*nums[i], dp_min[i-1]*nums[i], nums[i])

dp_min[i] = min(dp_max[i-1]*nums[i], dp_min[i-1]*nums[i], nums[i])

由于第i个状态只和i-1状态有关,所以可以考虑使用滚动数组的思想节省空间。

最终时间复杂度:O(n) 空间复杂度:O(1)

class Solution {
public:
    int maxProduct(vector<int>& nums) {
        if (nums.size() == 0) return 0;
        int pre_max = nums[0], pre_min = nums[0], ans = nums[0];
        for (int i = 1; i < nums.size(); i++) {
            int num = nums[i];
            // 注意:pre_max和pre_min要并行计算,所以需要用临时变量先存一下
            int last_max = pre_max, last_min = pre_min;
            pre_max = max(num, max(num*last_max, num*last_min));
            pre_min = min(num, min(num*last_max, num*last_min));
            ans = max(ans, pre_max);
        }
        return ans;
    }
};