907LeetCode子数组的最小数之和

最新推荐文章于 2024-07-25 23:26:35 发布

禮樂_

最新推荐文章于 2024-07-25 23:26:35 发布

阅读量349

点赞数

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

本文链接：https://blog.csdn.net/qq_44238580/article/details/127573820

版权

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

21 篇文章 0 订阅

订阅专栏

题目描述：

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

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

示例一：

输入：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。

示例二：

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

思路：

从左向右遍历数组并维护一个单调递增的栈，如果栈顶的元素大于等于当前元素 arr[i]则弹出栈，此时栈顶的元素即为左边第一个小于小于当前值的元素；

我们求出以当前值为最右且最小的子序列的长度 kkk，根据上述递推公式求出 dp[i]，最终的返回值即为 $\sum_{i=0}^{n-1}dp[i]$ 。

代码：

class Solution {
public:
    int sumSubarrayMins(vector<int>& arr) {
        int n = arr.size();
        long long sum = 0;
        long long mod = 1e9 + 7;
        stack<int> monoStack;
        vector<int> dp(n);
        for (int i = 0; i < n; i++) {
            while (!monoStack.empty() && arr[monoStack.top()] > arr[i]) {
                monoStack.pop();
            }
            int k = monoStack.empty() ? (i + 1) : (i - monoStack.top());
            dp[i] = k * arr[i] + (monoStack.empty() ? 0 : dp[i - k]);
            sum = (sum + dp[i]) % mod;
            monoStack.emplace(i);
        }
        return sum;
    }
};