- Leetcode 152. 乘积最大子数组
- 题目
- 给你一个整数数组 nums ,请你找出数组中乘积最大的非空连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
- 测试用例的答案是一个 32-位 整数。
- 子数组 是数组的连续子序列。
- 1 <= nums.length <= 2 * 10^4
- -10 <= nums[i] <= 10
- nums 的任何前缀或后缀的乘积都 保证 是一个 32-位 整数
- 解法
- 动态规划:定义 dp[n][2] 分别代表后缀的最小连续乘积与最大连续乘积,即 dp[i][0]:以 i 为后缀的最小连续乘积(负数乘最小负数可得最大正数),dp[i][1]:以 i 为后缀的最大连续乘积,初始化:当 i = 0 时 dp[0][0]=dp[0][1]=num[0],动规转移方程:
- dp[i][0] = min(num[i], dp[i-1][0]*num[i], dp[i-1][1]*num[i]),dp[i][1] = max(num[i], dp[i-1][0]*num[i], dp[i-1][1]*num[i])
- 最大值为 max(dp[0…n-1][1])
- 空间压缩:dp 数组中 i 仅与 i-1 相关,所以可用滚动数组 dp[2][2]
- 时间复杂度:O(n),空间复杂度:O(1)
- 代码
private int solution(int[] nums) {
if (nums == null || nums.length <= 0) {
return 0;
}
int len = nums.length;
int[][] dp = new int[2][2];
dp[0][0] = nums[0];
dp[0][1] = nums[0];
int res = nums[0];
for (int i = 1; i < len; i++) {
dp[i & 1][0] = Math.min(nums[i], Math.min(dp[(i + 1) & 1][0] * nums[i], dp[(i + 1) & 1][1] * nums[i]));
dp[i & 1][1] = Math.max(nums[i], Math.max(dp[(i + 1) & 1][0] * nums[i], dp[(i + 1) & 1][1] * nums[i]));
res = Math.max(res, dp[i & 1][1]);
}
return res;
}