网站首页 文章专栏 乘积最大子数组
给你一个整数数组 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) { if(nums.size() == 1) return nums[0]; int max_val = 1, min_val = 1; int res = INT_MIN; for(int i = 0; i < nums.size(); i++){ if(nums[i] < 0) swap(max_val, min_val); max_val = max(max_val * nums[i], nums[i]); min_val = min(min_val * nums[i], nums[i]); res = max(res, max_val); } return res; } };
另外,该题最长子序列的问题颇为相似,或许可以为此类问题考虑一个通用解法。