Letcode 5752. 子数组最小乘积的最大值
一个数组的 最小乘积 定义为这个数组中 最小值 乘以 数组的 和 。
比方说,数组 [3,2,5] (最小值是 2)的最小乘积为 2 * (3+2+5) = 2 * 10 = 20 。
给你一个正整数数组 nums ,请你返回 nums 任意 非空子数组 的最小乘积 的 最大值 。由于答案可能很大,请你返回答案对 10^9 + 7 取余 的结果。
请注意,最小乘积的最大值考虑的是取余操作 之前 的结果。题目保证最小乘积的最大值在 不取余 的情况下可以用 64 位有符号整数 保存。
子数组 定义为一个数组的 连续 部分。
示例 1:
输入:nums = [1,2,3,2]
输出:14
解释:最小乘积的最大值由子数组 [2,3,2] (最小值是 2)得到。
2 * (2+3+2) = 2 * 7 = 14
提示:
1 <= nums.length <= 10^5
1 <= nums[i] <= 10^7
没想到,各种双指针没想到,其他也没什么思路。其实关键是双指针的时候想的都是穷举左或者右端点,穷举某一个端点又无法保证最小值是固定的,所以很难处理。
这里既然是区间最小值乘以区间和,说明区间是被最小值限制着,我们可以去穷举这个最小值。以nums[i]为最小值的区间其实也不难想怎么去找,当一个区间的最小值已经固定为nums[i]的时候,显然我们要让这个区间越长越好。所以只要找到nums[i]左边/右边第一个比nums[i]更小的元素即可!这不就是单调栈吗,所以思路就出来了。
- r[i] l[i]表示nums[i]右边/左边第一个比nums[i]更小的元素索引,如果没有,就是n/-1
- 区间和用前缀和预处理一下
class Solution {
public:
int maxSumMinProduct(vector<int>& nums) {
int mod = 1e9 + 7, n = nums.size();
int r[n], l[n];
long long res = 0;
for (int i = 0; i < n; i++) r[i] = n, l[i] = -1;
stack<int> s;
long long spre[n + 1];
for (int i = 0; i < n; i++) {
while (!s.empty() && nums[i] < nums[s.top()]) {
r[s.top()] = i;
s.pop();
}
s.push(i);
}
while (!s.empty()) s.pop();
for (int i = n - 1; i >= 0; i--) {
while (!s.empty() && nums[i] < nums[s.top()]) {
l[s.top()] = i;
s.pop();
}
s.push(i);
}
spre[0] = 0;
for (int i = 1; i <= n; i++) spre[i] = spre[i - 1] + nums[i - 1];
for (int i = 0; i < n; i++) {
res = max(res, (long long)nums[i] * (spre[r[i]] - spre[l[i] + 1]));
}
return res % mod;
}
};