给定一个整数数组 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
def sumSubarrayMins(self, arr):
"""
:type arr: List[int]
:rtype: int
#单调栈:维护从大到小的排序(升序或降序)
#判断当前值与顶端的数相比较,小的话弹出来,入栈的时候是当前值
#dp(动态规划)
"""
# i = 4: 3 1 2 4 , 1 2 4, 2 4, 4 #数组元素i取值为4时,包含的子数组
# i = 3: 3 1 2, 1 2, 2
# i = 2: 3 1, 1
# i = 1: 3
# i = 0: 0
#[3 1 2 4] [3 1 2]的最小值是一样的,[1 2 4] [1 2]的最小值是一样的。当前元素放在前边比它小的数组里,不会影响前边比它小的元素里的最小值。
arr = [0] + arr
# 存储元素,下标从1开始
stack = [0]
# 到I子数组的最小值之和,找比当前元素最小的下标
dp = [0] * len(arr)
for i in range(len(arr)):
while arr[stack[-1]] > arr[i]:
stack.pop()
j = stack[-1]
dp[i] = dp[j] + (i-j)*arr[i]
stack.append(i)
return sum(dp) % (10 ** 9 + 7)