1. 程式人生 > 其它 >乘積最大陣列

乘積最大陣列

題目連結:https://leetcode-cn.com/problems/maximum-product-subarray
題目描述:
給你一個整數陣列 nums,請你找出陣列中乘積最大的連續子陣列(該子陣列中至少包含一個數字),並返回該子陣列所對應的乘積。

示例 1:
輸入: [2,3,-2,4]
輸出: 6
解釋:子陣列 [2,3] 有最大乘積 6。

示例 2:
輸入: [-2,0,-1]
輸出: 0
解釋:結果不能為 2, 因為 [-2,-1] 不是子陣列。

題解:


class Solution {
public:
    int maxProduct(vector<int>& nums) {
        //dp[i]:前i個數的最大乘積為dp[i]
        vector<int> dpMax(nums.size(), 0);     //記錄最大正乘積
        vector<int> dpMin(nums.size(), 0);     //記錄最小負乘積
        dpMax[0] = nums[0];
        dpMin[0] = nums[0];
        for(int i = 1; i < nums.size(); i++)
        {
            dpMax[i] = max(dpMax[i - 1] * nums[i], max(dpMin[i - 1] * nums[i], nums[i]));
            dpMin[i] = min(dpMin[i - 1] * nums[i], min(dpMax[i - 1] * nums[i], nums[i]));
        }
        int ans = dpMax[0];
        for(int i = 1; i < dpMax.size(); i++)
        {
           ans = max(ans, dpMax[i]);
        }
        return ans;

    }
};