LeetCode每日一题10.28

最新推荐文章于 2024-06-28 09:48:33 发布

TTSN_

最新推荐文章于 2024-06-28 09:48:33 发布

阅读量365

点赞数

分类专栏： LeetCode每日一题文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/weixin_45841172/article/details/127581118

版权

LeetCode每日一题专栏收录该内容

20 篇文章 0 订阅

订阅专栏

题目描述：

子数组的最小值之和

给定一个整数数组 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

代码：

class Solution {
public:
    int sumSubarrayMins(vector<int>& arr) {
        int MOD = (int)(1e9 + 7);
        int n = (int)arr.size();

        int left[n];
        int right[n];
        stack<int> stk;
        for (int i = 0; i < n; i ++){
            while (!stk.empty() && arr[stk.top()] >= arr[i]){
                stk.pop();
            }
            left[i] = (!stk.empty() ? stk.top() : -1);
            stk.push(i);
        }
        while(!stk.empty()){
            stk.pop();
        }
        for (int i = n - 1; i > -1; i --){
            while (!stk.empty() && arr[i] < arr[stk.top()]){
                stk.pop();
            }
            right[i] = (!stk.empty() ? stk.top() : n);
            stk.push(i);
        }

        long long res = 0;
        for (int i = 0; i < n; i ++){
            int L = i - left[i];
            int R = right[i] - i;
            res += (long long)L * R * arr[i] % MOD;
            res %= MOD;
        }
        return res;
    }
};