leetcode#152 乘积最大子数组
题目
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
示例:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
思路:
可以想到,对于第i个数,有两种选择,一是和归入前面的子数组,二是从当前位置开始重新开一个子数组,两者取最大即可。
其中能包含负数,对于当前位置想要取得最大值,要考虑正负两种情况,如果当前数为正数,那么我们希望他前面的数越大越好,也就是乘他前面的max,如果他是负数,我们希望它前面的数越小越好也就是min,所以维护两个值max、min来保证当前位置的最优解。
因为当前位置的值只与前一个元素有关,那么我们可以进行空间优化。
代码:
class Solution
{
public:
int maxProduct(vector<int> &nums)
{
int n = nums.size(),ans,dpmax,dpmin;
ans = dpmin = dpmax = nums[0];
for (int i = 1; i < n; ++i)
{
int mx=dpmax,mn=dpmin;
dpmax = max(mx * nums[i], max(mn * nums[i], nums[i]));
dpmin = min(mn * nums[i], min(mx * nums[i], nums[i]));
ans = max(ans, max(dpmax, dpmin));
}
return ans;
}
};