20190604-Leetcode-152. 乘积最大子序列

Leetcode-152. 乘积最大子序列

给定一个整数数组 nums ，找出一个序列中乘积最大的连续子序列（该序列至少包含一个数）。

示例 1:

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

示例 2:

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

思路：
    分别从前往后和从后向前寻找最大值，遇到0，则再开始下一个数的时候前面的乘积变为1。

C++ code：

class Solution {
public:
    
    int maxProduct(vector<int>& nums) {
        int n = nums.size();
        vector<int> left_max;
        //数组从前往后
        left_max.assign(nums.begin(), nums.end());
        vector<int> right_max;
        //数组从后向前
        right_max.assign(nums.rbegin(), nums.rend());
        for(int i = 1; i < n; i++){
        	//分别记录数组从前往后的乘积和从后向前的乘积
            //如果第i-1的位置为0， 则下一个数从自己开始。
            if(left_max[i - 1] != 0){
                left_max[i] *= left_max[i - 1];
            }
            if(right_max[i - 1] != 0){
                right_max[i] *= right_max[i - 1];
            }
        }
        //取两个数组中记录的最大值
        int l_maxPosition = max_element(left_max.begin(), left_max.end()) - left_max.begin();
        int r_maxPosition = max_element(right_max.begin(), right_max.end()) - right_max.begin();
        return max(left_max[l_maxPosition], right_max[r_maxPosition]);
        
    }
};

python code：

class Solution:
    def maxProduct(self, nums: List[int]) -> int:
        nums_left = copy.deepcopy(nums)
        nums_right = copy.deepcopy(nums[::-1])
        
        for i in range(1, len(nums)):
            # 这里or 1 是先判断第i - 1个位置是否为0，若是为0，则返回1，否则，返回自身。
            nums_left[i] *= nums_left[i - 1] or 1
            nums_right[i] *= nums_right[i - 1] or 1
        return max(max(nums_left), max(nums_right))