https://leetcode-cn.com/problems/maximum-product-subarray/description/
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
示例 1:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例 2:
输入: [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
类似于"53.最大子序和",不同的是需记录以当前元素结束的乘积最小值, 因为最小值乘一个负数可能翻身变最大值。
#define MAX(a, b) ((a) > (b) ? (a) : (b))
#define MIN(a, b) ((a) < (b) ? (a) : (b))
int maxProduct(int* nums, int numsSize){
int *dp = (int*)malloc(sizeof(int)*numsSize);
int max_product = nums[0];
int min_product = nums[0];
dp[0] = nums[0];
for(int i=1; i<numsSize; ++i){
// 类似于"53.最大子序和", dp[i]表示以i为终止下标的子数组乘积的最大值,问题的解即dp中的最大值
// ***不同的是需记录以当前元素结束的乘积最小值, 因为最小值乘一个负数可能翻身变最大值, min_product表示和dp[i]相对的最小值
// 动态转移方程也有所变化, 分下面四种情况
// 若dp[i-1]<0, array[i]>0,dp[i]=array[i]
// 若dp[i-1]<0, array[i]<0, dp[i]=max(dp[i-1]*array[i], min_product*array[i])
// 若dp[i-1]>0, array[i]<0, dp[i]=max(dp[i-1]*array[i], array[i], min_product*array[i])
// 若dp[i-1]>0, array[i]>0, dp[i]=dp[i-1]*array[i]
dp[i] = MAX(MAX(dp[i-1]*nums[i], min_product*nums[i]), nums[i]);
min_product = MIN(MIN(dp[i-1]*nums[i], min_product*nums[i]), nums[i]);
max_product = MAX(max_product, dp[i]);
}
if(dp){
free(dp);
}
return max_product;
}
压缩空间:
#define MAX(a, b) ((a) > (b) ? (a) : (b))
#define MIN(a, b) ((a) < (b) ? (a) : (b))
int maxProduct(int* nums, int numsSize){
// 压缩了一维dp
int res = nums[0];
int max_product = 1;
int min_product = 1;
for(int i=0; i<numsSize; ++i){
// 类似于"53.最大子序和", max_product表示以当前元素结束的子数组乘积的最大值
// 不同的是需记录以当前元素结束的乘积最小值, 用min_product表示, 因为最小值乘一个负数可能翻身变最大值
int tmp1 = max_product*nums[i];
int tmp2 = min_product*nums[i];
max_product = MAX(MAX(tmp1, tmp2), nums[i]);
min_product = MIN(MIN(tmp1, tmp2), nums[i]);
res = MAX(res, max_product);
}
return res;
}