Leetcode 152. 乘积最大子数组
题目
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
测试样例
示例 1:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例 2:
输入: [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
题解
动态规划
对于已知的连续数组,假设之前的是正数,现在的数是正数,那么我们就是之前最大的数X现在的数;如果现在的数是负数,那么我们就是之前最小的数X现在的数。假设之前的数是负数,现在的数是正数,那么我们就是之前最大的数X现在的数;如果现在的数是负数,那么就是之前最小的数X现在的数。
经过上述分析,我们就知道我们需要保存之间的最大连续乘和最小连续乘,我们的结果就是这两个结果乘现在的数的最大值。
我们令maxDp[i]表示nums的前i+1个数以nums[i]结尾的乘积最大值,minDp[i]表示nums的前i+1个数以nums[i]结尾的乘积最小值,dp[i]为nums前i+1个数中乘积最大的连续子区间,则转移方程如下
- maxDp[i] = max(maxDp[i-1]*nums[i],max(nums[i],minDp[i-1]*nums[i])),也就是说有三种情况,若maxDp[i-1]、nums[i]都是正数,则最大就是maxDp[i-1]*nums[i];若minDp[i-1]、nums[i]都是负数,则最大就是minDp[i-1]*nums[i];其他情况,就是nums[i]
- minDp[i] = min(minDp[i-1]*nums[i],min(nums[i],maxDp[i-1]*nums[i]))
- dp[i] = max(dp[i-1],maxDp[i])
初始条件,nums只有一个数时,显然,maxDp[0]=minDp[0]=dp[0]=nums[0]
代码
int maxProduct(vector<int>& nums) {
int n=nums.size();
vector<int> maxDp(n,0);
vector<int> minDp(n,0);
vector<int> dp(n,0);
maxDp[0] = nums[0];
minDp[0] = nums[0];
dp[0] = nums[0];
for(int i=1; i<n; i++){
maxDp[i] = max(maxDp[i-1]*nums[i],max(nums[i],minDp[i-1]*nums[i]));
minDp[i] = min(minDp[i-1]*nums[i],min(nums[i],maxDp[i-1]*nums[i]));
dp[i] = max(dp[i-1],maxDp[i]);
}
return dp[n-1];
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-product-subarray
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。