LeetCode-152-乘积最大子数组

Posted 樱花小猪

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode-152-乘积最大子数组相关的知识,希望对你有一定的参考价值。

LeetCode-152-乘积最大子数组

题目

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

示例 1:

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

示例 2:

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

思路

这道题乍一看很简单,和之前最最大子数组之和一样,我很轻松的列出了状态转移方程:

[dp[i]=max(dp[i-1]*x,x); ]

提交之后发现在自己错的离谱,问题就出在正数和负数上,这里的负数与负数相乘会得到更大的正数;

也就是说如果在前一个点保存的最大正数,当后面为负数的时候,正数反而没有用了,需要负数来相乘;

一开始我没有意识到问题的复杂性,自己做了一堆if语句来尝试解决这个问题,结果只能通过前面的简单的用例;

后面复杂的就不能通过了;

实在没有办法了,写了一个O(N^2)的暴力代码,结果当输入数据过大的时候超时了;

万般无奈就去看了题解,发现我就是题解中说明的典型错误示范。。。题解的正确思路是这样的:

我们可以根据正负性进行分类讨论。

考虑当前位置如果是一个负数的话,那么我们希望以它前一个位置结尾的某个段的积也是个负数,这样就可以负负得正,并且我们希望这个积尽可能「负得更多」,即尽可能小。如果当前位置是一个正数的话,我们更希望以它前一个位置结尾的某个段的积也是个正数,并且希望它尽可能地大。于是这里我们可以再维护一个 fmin?(i),它表示以第 i 个元素结尾的乘积最小子数组的乘积,那么我们可以得到这样的动态规划转移方程:

[egin{array}{l}{f_{max }}(i) = mathop {max }limits_{i = 1}^n { {f_{max }}(i - 1) imes {a_i},{f_{min }}(i - 1) imes {a_i},ai} \{f_{min }}(i) = mathop {max }limits_{i = 1}^n { {f_{max }}(i - 1) imes {a_i},{f_{min }}(i - 1) imes {a_i},ai} end{array} ]

看了人家的思路,顺便就看到了代码,果然还是官方的思路厉害,感觉自己还是太菜了。。

代码

class Solution {
public:
    int maxProduct(vector<int>& nums) {
        vector <int> maxF(nums), minF(nums);
        for (int i = 1; i < nums.size(); ++i) {
            maxF[i] = max(maxF[i - 1] * nums[i], max(nums[i], minF[i - 1] * nums[i]));
            minF[i] = min(minF[i - 1] * nums[i], min(nums[i], maxF[i - 1] * nums[i]));
        }
        return *max_element(maxF.begin(), maxF.end());
    }
};

以上是关于LeetCode-152-乘积最大子数组的主要内容,如果未能解决你的问题,请参考以下文章

LeetCode-152-乘积最大子数组

LeetCode-152-乘积最大子数组

LeetCode 152. 乘积最大子数组

LeetCode 152. 乘积最大子数组

LeetCode#152-乘积最大子数组-前缀和扩展到前缀积

[LeetCode] 152. 乘积最大子数组 ☆☆☆(动态规划)