题目如下
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
示例 1:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例 2:
输入: [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
解题方法1
此题要求找到连续乘积最大的子数组,连续数组的意义是什么,一个最直观的想法就是,如果前面的乘积在乘了元素i之后变大了,那么就把元素i加入到子数组中,否则,则从第i项开始,再进行记录。
但是这会遇到一个问题,正数如果乘以负数,那么就是变小了,此时如果抛弃掉前面的项,很明显是不对的,因为后面如果又遇到一个负数,那么最小值又变成最大值了,因此我们不能因为乘以了一个负数,就单纯抛弃前面的项。
此时问题似乎变复杂了,但是最大乘积只可能是最小值乘以负数,或者最大值乘以正数得到,因此我们维护一个最小值序列与最大值序列即可,这个状态转移就简单了,就是到第i个项的时候,最大值是从前i-1项的最大乘积得到,还是最小乘积得到,当然,也可能是当前元素,比如前面乘积最大值是负数,突然来了一个正数,那此时最大乘积数值就是它。
#include<iostream>
#include<vector>
using namespace std;
int maxProduct(vector<int>& nums)
{
int len = nums.size();
if (len == 0)
{
return 0;
}
vector<int> maxDp(len, 0);
vector<int> minDp(len, 0);
int res = nums[0];
maxDp[0] = nums[0];
minDp[0] = nums[0];
for (int i = 1; i < len; ++i)
{
minDp[i] = min(minDp[i - 1] * nums[i], maxDp[i - 1] * nums[i]);
maxDp[i] = max(maxDp[i - 1] * nums[i], minDp[i - 1] * nums[i]);
res = max(res, maxDp[i]);
}
return res;
}
int main()
{
vector<int>vec1 = { 2,3,-2,4 };
vector<int>vec2 = { -2,0,-1 };
cout << maxProduct(vec1) << endl;
cout << maxProduct(vec2) << endl;
return 0;
}
解题方法2
我们可以发现,其实不需要开两个数组来存储max和min,因为这两个数字可以相互转换,当遇到负数的时候,最大正数乘积可能会被转换成最小负数乘积,而最小负数乘积就可能会变成最大正数乘积, 而最大乘积也只可能在这两种情况下得到。
举个例子,对于[1,2,3,4,-1,-2]来说,最小乘积数组是[1,2,3,4,-1],最大则是[1,2,3,4,-1,-2],是由最小乘积得到的,但是由最小乘积转换来的正数不一定是最大乘积,比如[1,2,3,4,-1,-2, -1, 1,2,3,4,5],是由[-2,-1,1,2,3,4,5]数组得到的。
那么这样一来,我们还是需要额外利用一个max来记录过程中的最大值,因为最大乘积只可能从最小乘积的下一个状态和最大乘积中获取。
#include<iostream>
#include<vector>
using namespace std;
int maxProduct(vector<int>&nums)
{
int max = nums[0];
int imin = nums[0], imax = nums[0];
int n = nums.size();
for (int i = 1; i < n; i++)
{
if (nums[i] < 0)
{
imin = imax ^ imin;
imax ^= imin;
imin ^= imax;
}
imax = std::max(nums[i], imax * nums[i]);
imin = std::min(nums[i], imin * nums[i]);
max = std::max(max, imax);
}
return max;
}
int main()
{
vector<int>vec1 = { 2,3,-2,4 };
vector<int>vec2 = { -2,0,-1 };
cout << maxProduct(vec1) << endl;
cout << maxProduct(vec2) << endl;
return 0;
}