wy的leetcode刷题记录_Day26
时间:2022-10-28
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
思路
方法一:单调栈:读完题目后发现,本题要求一个数组的所有子数组的最小值之和,我认为可以将一个数组按区域最小值的辐射范围划分,按照划分的个数再乘以最小值的值,然后求和就可以得到答案。辐射范围由左边界有边界确定,首先我们遍历整个数组,对数组中的每一个元素这里称这为a[i]确定他所能辐射的范围(也就是在哪个区间他是最小值),对于左边界:我们向左寻找第一个小于等于a[i]的值,如果找到了那就记住其下标作为左边界。同理有边界也寻找第一个小于等于a[i]的值的下标。寻找方法我们介绍完了,这里我们介绍一下寻找的细节,首先我们从左往右遍历,如果对于a[i],维护一个栈,栈中的元素是其左边存在的子数组的最小值,如果a[i]>栈顶元素,那么a[i]的左边界无法向左扩充,如果a[i]<栈顶元素,那么就将栈顶元素出栈并且左边界向左扩充,如果a[i]他左边的值全部小于他本身的话,那么记左边界为left[i]=-1,此时只需要判断栈是否为空就可以知道是否左边界可以记为-1.同理有边界,最后计算公式就可以认为是:左边辐射范围右边辐射范围辐射值即(i-左边界)*(有边界-i)*a[i]。
代码
class Solution {
public:
int sumSubarrayMins(vector<int>& arr) {
int n = arr.size();
vector<int> monoStack;
vector<int> left(n), right(n);
for (int i = 0; i < n; i++) {
while (!monoStack.empty() && arr[i] <= arr[monoStack.back()]) {
monoStack.pop_back();
}
left[i] = i - (monoStack.empty() ? -1 : monoStack.back());
monoStack.emplace_back(i);
}
monoStack.clear();
for (int i = n - 1; i >= 0; i--) {
while (!monoStack.empty() && arr[i] < arr[monoStack.back()]) {
monoStack.pop_back();
}
right[i] = (monoStack.empty() ? n : monoStack.back()) - i;
monoStack.emplace_back(i);
}
long long ans = 0;
long long mod = 1e9 + 7;
for (int i = 0; i < n; i++) {
ans = (ans + (long long)left[i] * right[i] * arr[i]) % mod;
}
return ans;
}
};
收获
好难,单调栈是用来解决那种左右寻找第一个最小最大值的那种情况,也就是山峰山谷。