一个数组的 最小乘积 定义为这个数组中 最小值 乘以 数组的 和 。
- 比方说,数组
[3,2,5]
(最小值是2
)的最小乘积为2 * (3+2+5) = 2 * 10 = 20
。
给你一个正整数数组 nums
,请你返回 nums
任意 非空子数组 的最小乘积 的 最大值 。由于答案可能很大,请你返回答案对 109 + 7
取余 的结果。
请注意,最小乘积的最大值考虑的是取余操作 之前 的结果。题目保证最小乘积的最大值在 不取余 的情况下可以用 64 位有符号整数 保存。
子数组 定义为一个数组的 连续 部分。
输入: nums = [1,2,3,2] 输出: 14 解释: 最小乘积的最大值由子数组 [2,3,2] (最小值是 2)得到。 2 * (2+3+2) = 2 * 7 = 14 。
输入: nums = [2,3,3,1,2] 输出: 18 解释: 最小乘积的最大值由子数组 [3,3] (最小值是 3)得到。 3 * (3+3) = 3 * 6 = 18 。
输入: nums = [3,1,5,6,4,2] 输出: 60 解释: 最小乘积的最大值由子数组 [5,6,4] (最小值是 4)得到。 4 * (5+6+4) = 4 * 15 = 60 。
1 <= nums.length <= 105
1 <= nums[i] <= 107
impl Solution {
pub fn max_sum_min_product(nums: Vec<i32>) -> i32 {
let n = nums.len();
let mut prefix_sum = vec![0; n + 1];
let mut stack = vec![];
let mut indexl = vec![None; n];
let mut indexr = vec![None; n];
let mut ret = nums[0] as i64 * nums[0] as i64;
for i in 0..n {
prefix_sum[i + 1] = prefix_sum[i] + nums[i] as i64;
while let Some(&j) = stack.last() {
if nums[j] >= nums[i] {
stack.pop();
} else {
indexl[i] = Some(j);
break;
}
}
stack.push(i);
}
stack.clear();
for i in (0..n).rev() {
while let Some(&j) = stack.last() {
if nums[j] >= nums[i] {
stack.pop();
} else {
indexr[i] = Some(j);
break;
}
}
stack.push(i);
let mut sum = prefix_sum[indexr[i].unwrap_or(n)];
if let Some(j) = indexl[i] {
sum -= prefix_sum[j + 1];
}
ret = ret.max(nums[i] as i64 * sum);
}
(ret % 1_000_000_007) as i32
}
}