美文网首页
907. Sum of Subarray Minimums

907. Sum of Subarray Minimums

作者: 想学会飞行的阿番 | 来源:发表于2018-10-13 01:02 被阅读17次

思路:A[i]是最低值的次数 = (左边连续的大于A[i]的个数+1)*(右边连续大于等于A[i]的个数+1)

class Solution {
public:
    int sumSubarrayMins(vector<int>& A) {
        long long sum = 0;
        int n = A.size();
        vector<int> l(n,-1),r(n,n),st;
        for(int i = 0;i<n;i++){
            while(!st.empty()&&A[st.back()]>A[i])st.pop_back();
            l[i] = st.empty()?-1:st.back();
            st.push_back(i);
        }
        st.clear();
        for(int i = n-1;i>-1;i--){
            while(!st.empty()&&A[st.back()]>=A[i])st.pop_back();//注意这里
            r[i] = st.empty()?n:st.back();
            st.push_back(i);
        }
        for(int i =0;i<n;i++)
        {
            sum += (long long)A[i] *(long long)(i-l[i])*(long long)(r[i]-i);
        }
        return sum%((long long)1e9+7);
    }
};

相关文章

网友评论

      本文标题:907. Sum of Subarray Minimums

      本文链接:https://www.haomeiwen.com/subject/xovnaftx.html