一个数组的 最小乘积 定义为这个数组中 最小值 乘以 数组的 和 。
- 比方说,数组 [3,2,5] (最小值是 2)的最小乘积为 2 * (3+2+5) = 2 * 10 = 20 。
给你一个正整数数组 nums ,请你返回 nums 任意 非空子数组 的最小乘积 的 最大值 。由于答案可能很大,请你返回答案对 109 + 7 取余 的结果。
请注意,最小乘积的最大值考虑的是取余操作 之前 的结果。题目保证最小乘积的最大值在 不取余 的情况下可以用 64 位有符号整数 保存。
子数组 定义为一个数组的 连续 部分。
示例 1:
输入:nums = [1,2,3,2]
输出:14
解释:最小乘积的最大值由子数组 [2,3,2] (最小值是 2)得到。
2 * (2+3+2) = 2 * 7 = 14 。
示例 2:
输入:nums = [2,3,3,1,2]
输出:18
解释:最小乘积的最大值由子数组 [3,3] (最小值是 3)得到。
3 * (3+3) = 3 * 6 = 18 。
示例 3:
输入:nums = [3,1,5,6,4,2]
输出:60
解释:最小乘积的最大值由子数组 [5,6,4] (最小值是 4)得到。
4 * (5+6+4) = 4 * 15 = 60 。
提示:
- 1 <= nums.length <= 10^5
- 1 <= nums[i] <= 10^7
方法一:单调栈
「最小乘积」的定义为「最小值」乘以「和」,由于「和」较难进行枚举,我们可以考虑枚举「最小值」。
1、求前缀和
2、将nums中的每个值存入单调递增栈中,(为简便,将索引入栈)
4、以当前栈顶元素x为中心,往左找不小于x 的值left 和 向右找不小于 x 的值 right,乘积为 x * (presum[right] - presum[left]) , 以栈顶元素为当前最小值进行枚举。
class Solution {
public:
int maxSumMinProduct(vector<int>& nums) {
//前哨兵,避免左边界溢出
nums.insert(nums.begin(), 0);
//右哨兵,使栈中每个元素都能够枚举
nums.push_back(0);
int n = nums.size();
const int mod = 1e9 + 7;
vector<long long> presum(n, 0); //前缀和
presum[0] = nums[0];
for(int i=1; i<n; i++){
presum[i] = presum[i-1] + nums[i];
}
//单调递增栈
vector<int> stack;
long long res = 0;
for(int i=0; i<n; i++){
while(!stack.empty() && nums[stack.back()] > nums[i]){
int cur = nums[stack.back()]; //枚举栈顶元素
stack.pop_back();
int left = stack.back(); //左边界(不包含)
int right = i - 1; //右边界(包含)
res = max(res, (presum[right] - presum[left]) * cur);
}
stack.push_back(i);
}
return res % mod;
}
};