给定一个整数数组 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 *
1 <= arr[i] <= 3 *
记录一下单调栈的使用方法:
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