目录
乘积最大子数组
题目介绍
给你一个整数数组 nums ,请你找出数组中乘积最大的非空连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
示例1:
输入: nums = [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例1:
输入: nums = [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
思路:
题目要求求出最大连续子数组的乘积,如果找出所以的子数组求乘积再比较的话,这样时间复杂度会达到O(n^2),显然是不行的。可以这样想,题目给的数组是nums,我们求出以下标index结尾连续序列最大乘积,将这些结果记录在一个数组中,最后遍历这个数组得出最大乘积,即我们要的结果,这个数组应该如何求呢?找到转移方程,有两种情况,一种是要前面的,也就是dp[i-1],还有一种是不要前面的,自己单独成为一个序列起始位置,如下:
1.dp[i] = nums[i] * dp[i-1]
2.dp[i] = nums[i]
// 二者取大则得出
dp[i] = max(nums[i] * dp[i-1], nums[i]);
根据这种关系能求出部分正确结果,在数组中的数都是正数情况下能够得出正确结果,但是数组中有负数,答案就不正确,如下:nums=[-2, 3, -4]
下标 | 比较 | 结果 |
---|---|---|
0 | -2 | -2 |
1 | max(-2*3, 3) | 3 |
2 | max(-4*3, -4) | -4 |
根据这样的算法得出结果应该是3,实际应该是24,因为这里有偶数个负数,经过奇数次负数相乘会得到最小值,偶数次能得到最大值。也就是一个正数最大值乘以一个负数会变成最小值,同理负数最小值乘以一个负数会变成最大值。所以我们应该同时记录每个位置的最大值和最小值,当遇到负数时,要变换最大值和最小值再做乘积。举个例子:nums=[-2,3,-4,-5]
下标 | 比较 | 最大值 | 最小值 |
---|---|---|---|
0 | -2 | -2 | -2 |
1 | max(-23, 3) min(-23,3) | 3 | -6 |
2 | 负数:交换最大值和最小值 max(-6*-4, -4) min(-4*3,-4) | 24 | -12 |
3 | 负数:交换最大值和最小值 max(-12*-5, -5) min(24*-5,-5) | 60 | -120 |
所以得出最终结果为60
代码实现:
class Solution {
public:
int maxProduct(vector<int>& nums) {
int size = nums.size();
// dp1[i] 代表以i结尾连续序列最大乘积
// dp2[i] 代表以i结尾连续序列最小乘积
vector<int> dp1(size), dp2(size);
dp1[0] = dp2[0] = nums[0];
int ans = nums[0];
for (int i = 1; i < size; ++i) {
// 最大值和最小值互换
if (nums[i] < 0) swap(dp1[i-1], dp2[i-1]);
dp1[i] = max(nums[i]*dp1[i-1], nums[i]);
dp2[i] = min(nums[i]*dp2[i-1], nums[i]);
// update
ans = max(dp1[i], ans);
}
return ans;
}
};
上面的代码可以优化空间,用两个变量存储当前最大值和最小值即可,如下:
class Solution {
public:
int maxProduct(vector<int>& nums) {
int size = nums.size();
int curmax = nums[0], curmin = nums[0];
int ans = nums[0];
for (int i = 1; i < size; ++i) {
// 最大值和最小值互换
if (nums[i] < 0) swap(curmax, curmin);
curmax = max(nums[i]*curmax, nums[i]);
curmin = min(nums[i]*curmin, nums[i]);
// update
ans = max(curmax, ans);
}
return ans;
}
};