leetcode 907. 子数组的最小值之和

给定一个整数数组 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

class Solution {
public:
    int sumSubarrayMins(vector& arr) {
        int n=arr.size();
        vectorvec(n);
        vectorvex(n);
        stacks;
        long long sum=0;
        for(int i=n-1;i>=0;i--)
        {
            int temp=arr[i];
            while(!s.empty()&&arr[s.top()]>temp)
                s.pop();
            vec[i]=s.empty()? n:s.top();
            s.push(i);
        }
        while(!s.empty())
            s.pop();
        for(int i=0;i=temp)
                s.pop();
            vex[i]=s.empty()? -1:s.top();
            s.push(i);
        }
        for(int i=0;i

判断第i个数对于最小值的贡献,即arr[i]有多少次被作为最小值计算。设从arr[i]开始往左数,遇到的第一个小于arr[i]的值为arr[j],往右数遇到的第一个是arr[k],那么arr[i]的贡献就是(i-j)*(k-i)。

计算arr[j]和arr[k]的算法是利用单调栈实现的。

最后不得不用long long类型来求和。

你可能感兴趣的:(leetcode错题本,leetcode,leetcode,算法,c++)