给定一个整数数组 nums
,找出一个序列中乘积最大的连续子序列(该序列至少包含一个数)。
示例 1:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例 2:
输入: [-2,0,-1] 输出: 0 解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
思路:这道题和连续子序列的和很像最大子序列和,但是有很多不同,因为是乘积,如果一个很大的数乘以负数,立马就变成很小的数,而且如果乘以0就变成0了,与最大和很不一样(最大和加上负数和0只是在最大值上下浮动),所以不能用那道题的思路,需要加一些技巧。这道题中我们维护两个二维数组mm,mx,那么最大的值只会在如下三种情况下产生:
mm[i-1]*nums[i],mx[i-1]*nums[i],nums[i]
所以我们每次都维持两个二维数组的更新,其中mx表示三个的最大值,mm表示三个的最小值。递推公式如下:
int mm_tmp = mm;
int mx_tmp = mx;
mx = max(max(mx_tmp*nums[i], nums[i]), mm_tmp*nums[i]);
mm = min(min(mx_tmp*nums[i], nums[i]), mm_tmp*nums[i]);
res = max(res, mx);
这里把二维数组优化到了O(1)的空间复杂度。
参考代码:
int maxProduct(vector<int>& nums) {
if (nums.size() == 0) {
return 0;
}
if (nums.size() == 1) {
return nums[0];
}
int mm = nums[0], mx = nums[0], res = nums[0];
for (int i = 1; i < nums.size(); i++) {
int mm_tmp = mm;
int mx_tmp = mx;
mx = max(max(mx_tmp*nums[i], nums[i]), mm_tmp*nums[i]);
mm = min(min(mx_tmp*nums[i], nums[i]), mm_tmp*nums[i]);
res = max(res, mx);
}
return res;
}