1. 程式人生 > >【LeetCode】152. 乘積最大子序列 結題報告 (C++)

【LeetCode】152. 乘積最大子序列 結題報告 (C++)

原題地址:https://leetcode-cn.com/problems/maximum-product-subarray/description/

題目描述:

給定一個整數陣列 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) {
        int f[nums.size()];
        int g[nums.size()];
        f[0] = nums[0];     //最大值
        g[0] = nums[0];     //最小值
        int ans = f[0];
        for (int i = 1; i < nums.size(); i++) {
            f[i] = max(f[i - 1] * nums[i], nums[i]);
            f[i] = max(g[i - 1] * nums[i], f[i]);
            g[i] = min(f[i - 1] * nums[i], nums[i]);
            g[i] = min(g[i - 1] * nums[i], g[i]);
            ans = max(ans, f[i]);
        }
        return ans;
    }
};