【Lintcode】1734. Sum of Subarray Minimums

题目地址：

https://www.lintcode.com/problem/sum-of-subarray-minimums/description

给定一个长$n$的数组$A$，其有$\frac{n(n+1)}{2}$个子数组，求所有子数组的最小数的总和。答案模$10^9+7$后返回。

思路是单调栈。开一个单调递增栈，里面存$A$中的数的下标，并且要保持下标对应的$A$的值单调递增。遍历$A$，当遍历到$A[i]$的时候，如果栈不空并且栈顶$t$对应的值$A[t]>A[i]$，那么我们就知道了$A[t]$作为最小值的最长的那个区间的左右端点，先pop掉$A[t]$，如果栈空了，说明左端点可以取到$0$，否则取栈顶加$1$（因为栈顶是$A[t]$左边第一个小于等于它的数的下标），而$A[i]$是$A[t]$右边第一个小于它的数，所以右端点可以取到$i-1$。设左端点取$l$右端点取$r$，则以$A[t]$为最小值的区间的数量就是$(t-l+1)(r-t+1)$这么多，所以答案只需要累加一下$A[t]\times (t-l+1)(r-t+1)$即可。代码如下：

import java.util.ArrayDeque;
import java.util.Deque;

public class Solution {
    /**
     * @param A: an array
     * @return: the sum of subarray minimums
     */
    public int sumSubarrayMins(int[] A) {
        // Write your code here.
        Deque<Integer> stack = new ArrayDeque<>();
        
        int res = 0, MOD = (int) (1E9 + 7);
        for (int i = 0; i < A.length; i++) {
            while (!stack.isEmpty() && A[stack.peek()] > A[i]) {
                int top = stack.pop();
                // 这里left指以A[top]为最小值的区间在其左取数的方案数，下面乘起来正是组合数学里的乘法原理
                int left = top - (stack.isEmpty() ? -1 : stack.peek()), right = i - top;
                res += A[top] * left * right;
                res %= MOD;
            }

            stack.push(i);
        }
    
        while (!stack.isEmpty()) {
            int top = stack.pop();
            int left = top - (stack.isEmpty() ? -1 : stack.peek()), right = A.length - top;
            res += A[top] * left * right;
            res %= MOD;
        }
        
        return res;
    }
}

时空复杂度$O(n)$。