leetcode 907. Sum of Subarray Minimums解题思路

题目大意是: 将各个连续子数组的最小值累加,求个和值.

总体解题思路:

(1) 求出每个元素,在所有子数组中成为最小值的数量.

(2)元素值 乘以 数量, 再累加求和.

详细解题思路:

(1)求出各个元素的左边界.

(2)求出各个元素的右边界.

(3)在左边界内取一个位置(C左 1) 乘以 在右边界内 取一个位置(C右 1).

(4)注意相等的位置不能重复计算.

实现代码如下:

class Solution {
public:
    int sumSubarrayMins(vector<int>& a) {
        int as=a.size();
        vector<int> left(as,-1),right(as,as),st;
        for(int i=0;i<as;i++)//取左边界
        {
            while(!st.empty() && a[st.back()]>a[i])
            {
                st.pop_back();
            }
            if(!st.empty())
            {
                left[i]=st.back();
            }
            st.emplace_back(i);
        }
        st.clear();
        for(int i=as-1;i>=0;i--)//取右边界
        {
            while(!st.empty() && a[st.back()]>=a[i])//相等只计算一次
            {
                st.pop_back();
            }
            if(!st.empty())
            {
                right[i]=st.back();
            }
            st.emplace_back(i);
        }
        long long rst=0;
        for(int i=0;i<as;i++)//每个元素的子数组数量累加
        {
            rst+=((i-left[i])*(right[i]-i)*a[i]);
        }
        return rst%1000000007;
    }
};