题目:LeetCode
给你一个整数数组 nums
,请你找出数组中乘积最大的非空连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
测试用例的答案是一个 32-位 整数。
子数组 是数组的连续子序列。
示例 1:
输入: nums = [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
复制代码
示例 2:
输入: nums = [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。
复制代码
提示:
- 1<=nums.length<=2∗1041 <= nums.length <= 2 * 10^41<=nums.length<=2∗104
-10 <= nums[i] <= 10
nums
的任何前缀或后缀的乘积都 保证 是一个 32-位 整数
解题思路
由题意求连续子数组的乘积最大值,每个元素可加入子数组也可不加入,又是求最优解,一看就需要用动态规划求解。
乘法要复杂一些,如果当前的子数组乘积是负数,新加入元素也是负的,乘在一起,负负得正,子数组的乘积就变大了,说明状态转移不单单与当前元素本身有关还与之前的子数组的乘积有关。
状态转移方程 从前面的讨论来看,状态不能单单用前子数组和当前元素的乘积来决定,要依当前元素的正与负,前子数组乘积正和负都 有关系。
记f(i)是到元素i为止的可能的子数组最大乘积,如何得到f(i)呢?
如果,当前元素[i]是正的,那么f(i-1)要是正的,且越大越好;而如果,当前元素是负的,f(i-1)要是负的,且越小越好。因此,不光需要最大乘积状态,还需要最不乘积状态。
fmax(i)是到i为止的可能的子数组最大乘积,fmin(i)是到i为止可能的子数组最小乘积
代码实现
public int maxProduct(int[] nums) {
int pmin = nums[0];
int pmax = nums[0];
int result = pmax;
for (int i = 1; i < nums.length; i++) {
int max = Math.max(pmax * nums[i], Math.max(pmin * nums[i], nums[i]));
int min = Math.min(pmax * nums[i], Math.min(pmin * nums[i], nums[i]));
result = Math.max(result, max);
pmin = min;
pmax = max;
}
return result;
}
复制代码
复杂度分析
- 空间复杂度:O(1)O(1)O(1)
- 时间复杂度:O(n)O(n)O(n)