每日一题——子数组的最小值之和

给定一个整数数组 arr，找到 min(b) 的总和，其中 b 的范围为 arr 的每个（连续）子数组。

由于答案可能很大，因此 返回答案模 10^9 + 7 。

示例 1：

输入：arr = [3,1,2,4]
输出：17
解释：
子数组为 [3]，[1]，[2]，[4]，[3,1]，[1,2]，[2,4]，[3,1,2]，[1,2,4]，[3,1,2,4]。 
最小值为 3，1，2，4，1，1，2，1，1，1，和为 17。

示例 2：

输入：arr = [11,81,94,43,3]
输出：444

提示：

• 1 <= arr.length <= 3 * 104
• 1 <= arr[i] <= 3 * 104

思路：

1.暴力法

初看这道题，想法就是只需要两个循环，找出arr的所有子数组，然后找出子数组的最小值，累加起来就可以了，于是有以下代码：

for(int i=0;i<len;i++){
    sum+=arr[i];
    int min=arr[i];
    for(int j=i+1;j<len;j++){
        if(arr[j]<min){
            min=arr[j];
            sum+=min;
        } else{
            sum+=min;
        }

    }
}

通过找以arr[i]为第一个元素的所有子数组，以及记录最小值。

但是这样做会超时。

只通过了79/87个测试用例，所以只能换种方法了.......

 2.单调栈

题目要求的是每个子数组的最小值之和，实际上相当于，对于每个元素 arr[i]，求以 arr[i] 为最小值的子数组的个数，然后乘以 arr[i]，最后求和。

因此，题目的重点转换为：求以 arr[i] 为最小值的子数组的个数。对于 arr[i]，我们找出其左边第一个小于 arr[i] 的位置 left[i]，右侧第一个小于等于 arr[i] 的位置 right[i]，则以 arr[i]为最小值的子数组的个数为 (i−left[i])×(right[i]−i)。

注意，这里为什么要求右侧第一个小于等于 arr[i]arr[i] 的位置 right[i]right[i]，而不是小于 arr[i]arr[i] 的位置呢？这是因为，如果是右侧第一个小于 arr[i]arr[i] 的位置 right[i]right[i]，则会导致重复计算。

我们可以举个例子来说明，对于以下数组：

下标为 3的元素大小为 2，左侧第一个小于 2 的元素下标为 0，如果我们求右侧第一个小于 2 的元素下标，可以得到下标为 7。也即是说，子数组区间为 (0, 7)。注意，这里是开区间。

0 4 3 2 5 3 2 1
*     ^       *

按照同样的方法，我们可以求出下标为 6的元素的子数组区间，可以发现，其子数组区间也为 (0, 7)，也即是说，下标为 3 和下标为 6 的元素的子数组区间是重复的。这样就造成了重复计算。

0 4 3 2 5 3 2 1
*           ^ *

 
如果我们求的是右侧第一个小于等于其值的下标，就不会有重复问题，因为下标为 3 的子数组区间变为 (0, 6)，下标为 6 的子数组区间为 (0, 7)，两者不重复。

回到这道题上，我们只需要遍历数组，对于每个元素 arr[i]，利用单调栈求出其左侧第一个小于 arr[i]的位置 left[i]，右侧第一个小于等于 arr[i]的位置 right[i]，则以 arr[i]为最小值的子数组的个数为 (i−left[i])×(right[i]−i)，然后乘以 arr[i]，最后求和即可。

注意数据的溢出以及取模操作。

时间复杂度 O(n)，其中 n表示数组 arr 的长度。

using ll = long long;
const int mod = 1e9 + 7;

class Solution {
public:
    int sumSubarrayMins(vector<int>& arr) {
        int n = arr.size();
        vector<int> left(n, -1);
        vector<int> right(n, n);
        stack<int> stk;
        for (int i = 0; i < n; ++i) {
            while (!stk.empty() && arr[stk.top()] >= arr[i]) stk.pop();
            if (!stk.empty()) left[i] = stk.top();
            stk.push(i);
        }
        stk = stack<int>();
        for (int i = n - 1; i >= 0; --i) {
            while (!stk.empty() && arr[stk.top()] > arr[i]) stk.pop();
            if (!stk.empty()) right[i] = stk.top();
            stk.push(i);
        }
        ll ans = 0;
        for (int i = 0; i < n; ++i) {
            ans += (ll)(i - left[i]) * (right[i] - i) * arr[i] % mod;
            ans %= mod;
        }
        return ans;
    }
};