【LeetCode热题100】【动态规划】乘积最大子数组

2024-04-20 11:05:59 浏览数 (2)

题目链接:152. 乘积最大子数组 - 力扣(LeetCode)

要找乘积最大的连续子数组,我们之前做过找和最大的连续子数组【LeetCode热题100】【普通数组】最大子数组和-CSDN博客

对于和来讲,定义dp[i]是以nums[i]为结尾的最大子数组的和,那么dp[i]要么就是nums[i]加上之前的dp[i-1],即前一个子数组加上当前元素,要么是nums[i]自己新开一个子数组,所以状态转移方程

代码语言:javascript复制
dp[i] = std::max(dp[i - 1]   nums[i], nums[i]);

但是乘积与和不一样的地方在于负数的情况,负数加负数的和是越来越小的,但是乘积会因为碰到一个负数变成最大或者变成最小,因此需要用两个数组记录状态,dpMin记录最小,dpMax记录最大

代码语言:javascript复制
class Solution {
public:
    int maxProduct(vector<int> &nums) {
        vector<int> dpMax(nums.size()), dpMin(nums.size());
        int ans = dpMax[0] = dpMin[0] = nums[0];
        for (int i = 1; i < nums.size();   i) {
            dpMax[i] = max(nums[i], max(nums[i] * dpMax[i - 1], nums[i] * dpMin[i - 1]));
            dpMin[i] = min(nums[i], min(nums[i] * dpMax[i - 1], nums[i] * dpMin[i - 1]));
            ans = max(dpMax[i], ans);
        }
        return ans;
    }
};

0 人点赞