LeetCode：907. 子数组的最小值之和（单调栈 C++ 、Java）

907. 子数组的最小值之和

题目描述：

实现代码与解析：

单调栈

原理思路：

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 * 104
1 <= arr[i] <= 3 * 104

实现代码与解析：

单调栈

C++

class Solution {
public:int sumSubarrayMins(vector<int>& arr) {int mod = 1e9 + 7;long res = 0;int n = arr.size();vector<int> left(n, 0), right(n, 0);stack<int> stk, stk2;// 找左边最小for (int i = 0; i < n; i++) {while (stk.size() && arr[stk.top()] >= arr[i]) {stk.pop();}if (stk.size()) {left[i] = stk.top();} else {left[i] = -1;}stk.push(i);}// 找右边最小for (int i = n - 1; i >= 0; i--) {while (stk2.size() && arr[stk2.top()] > arr[i]) {stk2.pop();}if (stk2.size()) {right[i] = stk2.top();} else {right[i] = n;}stk2.push(i);}for (int i = 0; i < n; i++) {res += (long)(i - left[i]) * (right[i] - i) % mod * arr[i] % mod;}return res % mod;}
};

java

class Solution {private static final int MOD = 1000000007;public int sumSubarrayMins(int[] arr) {int n = arr.length;int[] left = new int[n];int[] right = new int[n];Deque<Integer> stk = new LinkedList<>();for (int i = 0; i < n; i++) {while (!stk.isEmpty() && arr[stk.peek()] > arr[i]) {stk.pop();}if (stk.isEmpty()) {left[i] = -1;} else {left[i] = stk.peek();}stk.push(i);}stk.clear();for (int i = n - 1; i >= 0; i--) {while (!stk.isEmpty() && arr[stk.peek()] >= arr[i]) {stk.pop();}if (stk.isEmpty()) {right[i] = n;} else {right[i] = stk.peek();}stk.push(i);}long res = 0;for (int i = 0; i < n; i++) {res += (long)(i - left[i]) * (right[i] - i) * arr[i];res %= MOD;}return (int)res;}
}