【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 * 10^{4}
  • 1 <= arr[i] <= 3 * 10^{4}

记录一下单调栈的使用方法:

class Solution:
    def sumSubarrayMins(self, arr: List[int]) -> int:
        mod=10**9+7
        n=len(arr)
        ans=0
        left=[-1]*n
        right=[n]*n
        stack=[]

        for i in range(n):
            while stack and arr[stack[-1]]>=arr[i]:
                stack.pop()
            if stack:
                left[i]=stack[-1] 
            stack.append(i)

        stack=[]

        for i in range(n-1,-1,-1):
            while stack and arr[stack[-1]]>arr[i]:
                stack.pop()
            if stack:
                right[i]=stack[-1] 
            stack.append(i)

        for i in range(n):
            ans+=(i-left[i])*(right[i]-i)*arr[i]
        return ans%mod

你可能感兴趣的:(刷题,leetcode,算法,数据结构)