题目描述
给定一个整数数组 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
算法分析
先求出对于某个位置i的左右边界
l数组为左边严格比arr[i]小的数的位置
r数组为右边非严格比arr[i]小的数的位置
最后遍历数组,计算每个数算了几次
代码
class Solution {
public:
int sumSubarrayMins(vector<int>& arr) {
int n = arr.size();
const int mod = 1e9+7;
vector<long long> l(n, -1), r(n, n);
stack<int> st;
for(int i =0; i <n; ++i) {
while(!st.empty() && arr[st.top()] >= arr[i]) {
r[st.top()] = i;
st.pop();
}
if(!st.empty()) {
l[i] = st.top();
}
st.push(i);
}
long long ans = 0;
for(int i= 0; i < n; ++i) {
ans += (i - l[i])*(r[i] - i) * arr[i] % mod;
}
return (ans + mod)%mod;
}
};