原题链接:乘积最大子数组
更好的解法
思路:
这里也许一开始会想到直接f[i] = max(f[i - 1]*nums[i], nums[i])
,但这里有问题,如果有序列{5,6,−3,4,−3}, 求出来的f[i]
数组为{5,30,−3,4,−3},最大值为30,然后这并不对,最大值是整个序列相加,为什么?我们可以看最后一个数字,因为前面所有的数字相乘为一个负数,然而最后一个数字为负数,最后结果相乘为正数!故我们可以得出,如果前面为负值,那么当前数如果为负数,也可能使得答案最大,所以我们需要借助两个状态表示,一个表示以i结尾的最大子序列,一个表示以i结尾的最小子序列
那么转移方程就为
f_max[i] = max(nums[i], max(f_max[i - 1] * nums[i], f_min[i - 1] * nums[i]));
f_min[i] = min(nums[i], min(f_max[i - 1] * nums[i], f_min[i - 1] * nums[i]));
时间复杂度: O ( n ) O(n) O(n)
代码:
class Solution {
public:
int maxProduct(vector<int>& nums) {
int n = nums.size();
vector<int> f_max(n), f_min(n);
for(int i = 0;i < n;i ++) {
if(i == 0) f_max[i] = f_min[i] = nums[i];
else {
f_max[i] = max(nums[i], max(f_max[i - 1] * nums[i], f_min[i - 1] * nums[i]));
f_min[i] = min(nums[i], min(f_max[i - 1] * nums[i], f_min[i - 1] * nums[i]));
}
}
return *max_element(f_max.begin(), f_max.end());
}
};