907. 子数组的最小值之和

霜白sakura

于 2022-10-28 19:46:42 发布

阅读量144

点赞数

分类专栏： leetcode 文章标签：算法 c++ leetcode

本文链接：https://blog.csdn.net/weixin_44458954/article/details/127577830

版权

leetcode 专栏收录该内容

58 篇文章 0 订阅

订阅专栏

题目：907. 子数组的最小值之和
思路：单调栈
在这里插入图片描述

class Solution {
public:
    int sumSubarrayMins(vector<int>& arr) {
        int MOD = 1e9+7;
        int n = arr.size();
        int ans = 0;
        vector<int> li(n,-1);
        vector<int> ri(n,n);
        stack<int> q;
        for(int i=0;i<n;++i){
            while(!q.empty() && arr[q.top()]>arr[i]){
                ri[q.top()]=i;
                q.pop();
            }
            q.push(i);
        }
        while(!q.empty()){
            q.pop();
        }
        for(int i=n-1;i>=0;i--){
            while(!q.empty() && arr[q.top()]>=arr[i]){
                li[q.top()]=i;
                q.pop();
            }
            q.push(i);
        }
        for(int i=0;i<n;i++){
            int a=i-li[i];
            int b=ri[i]-i;
            ans+=a*1L*b%MOD*arr[i]%MOD;
            ans%=MOD;
        }
        return ans;
    }
};