【LeetCode & 剑指offer刷题】动态规划与贪婪法题5:Maximum Product Subarray

【LeetCode & 剑指offer 刷题笔记】目录(持续更新中...)

Maximum Product Subarray

Given an integer array nums, find the contiguous subarray within an array (containing at least one number) which has the largest product.
Example 1:
Input: [2,3,-2,4]
Output: 6
Explanation: [2,3] has the largest product 6.
Example 2:
Input: [-2,0,-1]
Output: 0
Explanation: The result cannot be 2, because [-2,-1] is not a subarray.

C++
 
/*
问题:求最大子数组乘积
方法:动态规划
两个dp数组,其中f[i]和g[i]分别表示包含nums[i](以nums[i]结尾)时的最大和最小子数组乘积,
初始化时f[0]和g[0]都初始化为nums[0],其余都初始化为0。
 
从数组的第二个数字开始遍历,此时的最大值和最小值只会在这三个数字之间产生,
即f[i-1]*nums[i],g[i-1]*nums[i],和nums[i]。
所以我们用三者中的最大值来更新f[i],用最小值来更新g[i],然后用f[i]来更新结果res即可
*/
class Solution
{
public:
    int maxProduct(vector<int>& nums)
    {
        if(nums.empty()) return 0;
           
        int res = nums[0], n = nums.size();
        vector<int> f(n), g(n); //分配空间,初始化
        f[0] = nums[0];
        g[0] = nums[0];
       
        for (int i = 1; i < n; i++) //从a[1]开始遍历
        {
            f[i] = max(max(f[i - 1] * nums[i], g[i - 1] * nums[i]), nums[i]); //最大数更新
            g[i] = min(min(f[i - 1] * nums[i], g[i - 1] * nums[i]), nums[i]); //最小数更新
            res = max(res, f[i]); //结果更新
        }
        return res;
    }
};
 
/*
空间优化,用两个变量代替数组dp 
*/
class Solution
{
public:
    int maxProduct(vector<int>& nums)
    {
        if (nums.empty()) return 0;
        int res = nums[0], mn = nums[0], mx = nums[0];
        for (int i = 1; i < nums.size(); ++i)
        {
            int tmax = mx, tmin = mn;//上一次的最小值和最大值
            mx = max(max(nums[i], tmax * nums[i]), tmin * nums[i]);
            mn = min(min(nums[i], tmax * nums[i]), tmin * nums[i]);
            res = max(res, mx);
        }
        return res;
    }
};
 

 

posted @ 2019-01-06 15:37  wikiwen  阅读(125)  评论(0编辑  收藏  举报