动态规划|152. 乘积最大子数组
152. 乘积最大子数组 - 力扣(LeetCode) (leetcode-cn.com)
题目
给你一个整数数组 nums ,请你找出数组中乘积最大的非空连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
测试用例的答案是一个 32-位 整数。
子数组 是数组的连续子序列。
示例 1:
输入: nums = [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例 2:输入: nums = [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
解题思路
设置二维dp数组dp[2] [numsSize],其中
-
dp[0] [k]表示以nums数组下标为k的元素结尾的连续子数组的乘积最大值
-
dp[1] [k]表示以nums数组下标为k的元素结尾的连续子数组的乘积最小值
满足表达式
dp[0][k]=max{dp[0][k-1]*nums[k],nums[k]}
若dp[0] [k-1]*nums[k]>nums[k],则dp[0] [k-1]*nums[k]结果为以nums[k]结尾的连续子数组乘积最大值,否则另起炉灶,以nums[k]结尾的连续子数组乘积最大值就只有他自己一个(说白了就是:前一个连续子数组乘积最大值和自己相乘结果比自己还小,果断抛弃他们,自立家门)
dp[1][k]=min{dp[1][k-1]*nums[k],nums[k]}
与计算乘积最大同理
之所以要维护一个最小值,是因为如果当前数组值为负数,dp[0] [k-1]*nums[k]就可能是最小值(假设正数×负数的情况),dp[1] [k-1]*nums[k]就可能是最大值(假设负数×负数的情况),故当遇到数组值为负数时交换此时最大最小值
代码
int maxProduct(int* nums, int numsSize){
int min,max,MAXproduct;
int dp[2][numsSize];//dp[0][k]表示以nums数组下标为k的元素结尾的连续子数组的乘积最大值,dp[1][k]表示以nums数组下标为k的元素结尾的连续子数组的乘积最小值
dp[0][0]=nums[0];
dp[1][0]=nums[0];
max=dp[0][0];//max表示dp[0][k]的前一个下标结尾连续子数组的乘积最大值,即max=dp[0][k-1]
min=dp[1][0];//min表示dp[1][k]的前一个下标结尾连续子数组的乘积最小值,即min=dp[1][k-1]
MAXproduct=dp[0][0];//表示此时非空连续子数组乘积最大值
for(int i=1;i<numsSize;i++){
if(nums[i]<0){//如果当前数组值为负数,则交换max与min,因为最小值×负数可能是最大值,最大值×负数可能是最小值
int temp = max;
max=min;
min=temp;
}
dp[0][i]=max*nums[i]>nums[i]?max*nums[i]:nums[i];//若dp[0][k-1],即max乘当前数组值大于当前数组值,则乘积结果为以nums[i]结尾的连续子数组乘积最大值,否则另起炉灶,以nums[i]结尾的连续子数组乘积最大值就是他自己
dp[1][i]=min*nums[i]<nums[i]?min*nums[i]:nums[i];
max=dp[0][i];//在下个循环前更新前一个下标结尾连续子数组的乘积最大值
min=dp[1][i];
if(dp[0][i]>MAXproduct)
MAXproduct=dp[0][i];//此时非空连续子数组乘积最大值
}
return MAXproduct;
}