【乘积最大子数组】

题目来源:https://leetcode.cn/problems/maximum-product-subarray/


乘积最大子数组

题目介绍

给你一个整数数组 nums ,请你找出数组中乘积最大的非空连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。

示例1:

输入: nums = [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。

示例1:

输入: nums = [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。

思路:
题目要求求出最大连续子数组的乘积,如果找出所以的子数组求乘积再比较的话,这样时间复杂度会达到O(n^2),显然是不行的。可以这样想,题目给的数组是nums,我们求出以下标index结尾连续序列最大乘积,将这些结果记录在一个数组中,最后遍历这个数组得出最大乘积,即我们要的结果,这个数组应该如何求呢?找到转移方程,有两种情况,一种是要前面的,也就是dp[i-1],还有一种是不要前面的,自己单独成为一个序列起始位置,如下:

1.dp[i] = nums[i] * dp[i-1]
2.dp[i] = nums[i]
// 二者取大则得出
dp[i] = max(nums[i] * dp[i-1], nums[i]);

根据这种关系能求出部分正确结果,在数组中的数都是正数情况下能够得出正确结果,但是数组中有负数,答案就不正确,如下:nums=[-2, 3, -4]

下标比较结果
0-2-2
1max(-2*3, 3)3
2max(-4*3, -4)-4

根据这样的算法得出结果应该是3,实际应该是24,因为这里有偶数个负数,经过奇数次负数相乘会得到最小值,偶数次能得到最大值。也就是一个正数最大值乘以一个负数会变成最小值,同理负数最小值乘以一个负数会变成最大值。所以我们应该同时记录每个位置的最大值和最小值,当遇到负数时,要变换最大值和最小值再做乘积。举个例子:nums=[-2,3,-4,-5]

下标比较最大值最小值
0-2-2-2
1max(-23, 3) min(-23,3)3-6
2负数:交换最大值和最小值 max(-6*-4, -4) min(-4*3,-4)24-12
3负数:交换最大值和最小值 max(-12*-5, -5) min(24*-5,-5)60-120

所以得出最终结果为60

代码实现:

class Solution {
public:
    int maxProduct(vector<int>& nums) {
        int size = nums.size();
        // dp1[i] 代表以i结尾连续序列最大乘积
        // dp2[i] 代表以i结尾连续序列最小乘积
        vector<int> dp1(size), dp2(size);
        dp1[0] = dp2[0] = nums[0]; 
        int ans = nums[0];
        for (int i = 1; i < size; ++i) {
            // 最大值和最小值互换
            if (nums[i] < 0) swap(dp1[i-1], dp2[i-1]);
            dp1[i] = max(nums[i]*dp1[i-1], nums[i]);
            dp2[i] = min(nums[i]*dp2[i-1], nums[i]);
            // update
            ans = max(dp1[i], ans);
        }
        return ans;
    }
};

上面的代码可以优化空间,用两个变量存储当前最大值和最小值即可,如下:

class Solution {
public:
    int maxProduct(vector<int>& nums) {
        int size = nums.size();
        int curmax = nums[0], curmin = nums[0];
        int ans = nums[0];
        for (int i = 1; i < size; ++i) {
            // 最大值和最小值互换
            if (nums[i] < 0) swap(curmax, curmin);
            curmax = max(nums[i]*curmax, nums[i]);
            curmin = min(nums[i]*curmin, nums[i]);
            // update
            ans = max(curmax, ans);
        }
        return ans;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

呆呆兽学编程

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值