# 907. 子数组的最小值之和

给定一个整数数组 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 * 10^4$
  • 1 <= arr[i] <= $3 * 10^4$

# 题解

class Solution {
public:
int sumSubarrayMins(vector<int>& arr) {
int n = arr.size();
long long ans = 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]);
ans = (ans + dp[i]) % mod;
monoStack.emplace(i);
}
return ans;
}
};

复杂度分析

  • 时间复杂度:$O (n)$
  • 空间复杂度:$O (n)$