Leetcode.1856 子数组最小乘积的最大值

题目链接

Leetcode.1856 子数组最小乘积的最大值 Rating ： 2051

题目描述

一个数组的 最小乘积 定义为这个数组中 最小值 乘以 数组的 和 。

• 比方说，数组 $[3,2,5]$ （最小值是 2）的最小乘积为 $2\times (3+2+5)=2\times 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 。

示例 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$

解法：单调栈

$s$ 是 $nums$ 的前缀和。

对于 $nums$ 的每一个元素 $nums[i]$ ，我们考虑 以$nums[i]$ 为最小值的子数组 $nums[l,r]$，那么最小乘积为 $nums[i]\times (s[r]-s[l-1])$。

通过 单调栈 ，我们可以求出 $nums[i]$ 左边第一个 $<nums[i]$ 的下标 $l$ ；求出右边第一个 $\le nums[i]$ 的下标 $r$。所以以 $nums[i]$ 为最小值的子数组为 $nums(l,r]$，故最小乘积为 $nums[i]\times (s[r+1]-s[l+1])$，因为$nums$ 下标从 0 开始，而 $s$ 下标从 1 开始。

时间复杂度：

C++代码：

const int MOD = 1e9+7;
using LL = long long;

class Solution {
public:
    int maxSumMinProduct(vector<int>& nums) {
        int n = nums.size();
        vector<LL> s(n + 1);
        for(int i = 1;i <= n;i++) s[i] = s[i - 1] + nums[i - 1];
        
        //避免边界条件
        stack<int> stk{{-1}};
        nums.push_back(-1);

        LL ans = 0;

        for(int i = 0;i < nums.size();i++){
        //此时的 i 就是 nums[stk.top()] 的右边界 r
            while(stk.size() > 1 && nums[stk.top()] >= nums[i]){
                int k = stk.top();
                stk.pop();
                
                //最后一次弹出栈时，此时的 stk.top() 一定是 nums[k] 的左边界 l
                //故 最乘积为 nums[k] * (s[r] - s[l + 1])
                ans = max(ans , (s[i] - s[stk.top() + 1]) * nums[k]);
            }
            stk.push(i);
        }
        return ans%MOD;
    }
};