编程训练第一百零二期——乘积最大子数组

编程问题：

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

---

示例：

• 输入: [2,3,-2,4]
  输出: 6
  解释: 子数组 [2,3] 有最大乘积 6。
• 输入: [-2,0,-1]
  输出: 0
  解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。

---

解法：

1.动态规划
状态：考虑当前位置如果是一个负数的话，希望以它前一个位置结尾的某个段的积也是个负数，这样就可以负负得正，进一步希望这个积尽可能小。如果当前位置是一个正数的话，更希望以它前一个位置结尾的某个段的积也是个正数，并且希望它尽可能地大。因此dpMax[i] 表示 nums 前 i 个的元素最大的子数组乘积，dpMin[i]表示nums前 i 个的元素最小的子数组乘积。

转移方程： dpMax[i] = max(dpMax[i-1] * nums[i], max(dpMin[i-1] * nums[i], nums[i]));
dpMin[i] = min(dpMax[i-1] * nums[i], min(dpMin[i-1] * nums[i], nums[i]));。

初始化：dpMax(nums), dpMin(nums) ，初始化为原始数组。

结果：dpMax数组最大的元素

时间复杂度O(N)
空间复杂度O(N)

class Solution {
public:
    int maxProduct(vector<int>& nums) {
        vector<int> dpMax(nums), dpMin(nums);
        for(int i = 1; i < nums.size(); i++)
        {
            dpMax[i] = max(dpMax[i-1] * nums[i], max(dpMin[i-1] * nums[i], nums[i]));
            dpMin[i] = min(dpMax[i-1] * nums[i], min(dpMin[i-1] * nums[i], nums[i]));
        }
        sort(dpMax.begin(), dpMax.end(), greater<int>());
        return dpMax[0];
    }
};

压缩状态
时间复杂度O(N)
空间复杂度O(1)

class Solution {
public:
    int maxProduct(vector<int>& nums) {
        int dpMax = nums[0], dpMin = nums[0], res = nums[0];
        for(int i = 1; i < nums.size(); i++)
        {
            int tempMax = dpMax, tempMin = dpMin;
            dpMax = max(tempMax * nums[i], max(tempMin * nums[i], nums[i]));
            dpMin = min(tempMax * nums[i], min(tempMin * nums[i], nums[i]));
            res = max(res, dpMax);
        }
        
        return res;
    }
};