https://leetcode.com/problems/sum-of-subarray-minimums/description/
题目大意:求连续子序列中最小值的和。
解题思路:以每个位置向左和右查找子序列,子列的数目*A[i] 为答案,子列的数目为 左边能到的最远距离*右边最远的距离。
暴力查找会超时,可以使用栈来记录前序的情况,后序的情况也可以记录。
#define mod 1000000007
class Solution {
public:
int sumSubarrayMins(vector<int>& A) {
int n = A.size();
int res =0;
stack<int> st;
for(int i=0;i<n;i++)
{
while(!st.empty() && A[st.top()]>A[i])
st.pop();
int left = st.empty() ? -1:st.top();
st.push(i);
int j=i+1;
while(j<n && A[j]>=A[i]) j++;
res = res + ((i-left)*(j-i) % mod) *A[i] ;
res = res % mod;
}
return res;
}
};