907.子数组的最小值之和

原创已于 2023-11-27 17:46:43 修改 · 483 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #leetcode

于 2023-11-27 13:47:00 首次发布

leetcode 专栏收录该内容

16 篇文章

订阅专栏

单调栈 + 贡献法

每个子数组的最小值的和，即对于每个元素，包含该元素的所有子数组的个数与该元素值的乘积即为所求值
每个元素贡献的次数：以该元素为最小值的所有子数组的个数
区间法求符合条件的包含某元素的子数组个数：左右边界均为最靠近该元素的并且小于(等于)该元素的元素所处的位置
左右边界：leftBound, rightBound, 子数组个数：(i - leftBound) * (rightBound - i)
(1) 使用枚举查找最靠近该元素的并且小于(等于)该元素的元素所处的位置，超时。。。
(2) 使用单调栈，栈中保存下标，对应的元素值由小到大，以当前元素为最大值，pop掉栈中比当前元素大的值的下标，栈顶即为所要位置(边界)

class Solution {
public:
    int sumSubarrayMins(vector<int>& arr) {
        stack<int> s;
        int n = arr.size();
        vector<int> leftBound(n, -1);
        vector<int> rightBound(n, n);
        for(int i = 0; i < n; i++)
        {
            while(s.size() && arr[i] <= arr[s.top()])
                s.pop();
            leftBound[i] = s.empty() ? -1 : s.top();
            s.push(i);
        }
        s = {};
        for(int i = n - 1; i >= 0; i--)
        {
            while(s.size() && arr[i] < arr[s.top()])
                s.pop();
            rightBound[i] = s.empty() ? n : s.top();
            s.push(i);
        }
        long long ret = 0, mod = 1e9 + 7;
        for(int i = 0; i < n; i++)
        {
            ret = ( ret + 
            (long long)arr[i] * (i - leftBound[i]) * (rightBound[i] - i) ) % mod; 
        }
        return ret;
    }
};