力扣907.子数组的最小值之和

力扣907.子数组的最小值之和

• 考虑每个数对答案的贡献

  • 对于每个元素 找到左边界(严格小于)
    • 右边界(小于等于)
  • 这样出现重复元素也不会重复计算答案
  • 对于答案贡献为 arr[i] * (i - l) * (r - i)

•   class Solution {
        const int MOD = 1e9+7;
    public:
        int sumSubarrayMins(vector<int>& arr) {
            int n = arr.size();
            vector<int> l(n,-1),r(n,n);
            stack<int> st;
            for(int i=0;i<n;i++)
            {
                while(!st.empty() && arr[st.top()] >= arr[i]) st.pop();
                if(!st.empty()) l[i] = st.top();
                st.push(i);
            }
    
            st = stack<int>();
            for(int i=n-1;i>=0;i--)
            {
                while(!st.empty() && arr[st.top()] > arr[i]) st.pop();
                if(!st.empty()) r[i] = st.top();
                st.push(i);
            }
    
            long res=0L;
            for(int i=0;i<n;i++)
            {
                res += (long)arr[i] * (i - l[i]) * (r[i] - i);
            }
            return res % MOD;
        }
    };