力扣每日一题 子数组的最小值之和

Day 31

Date: October 28, 2022 10:08 AM
LinkedIn: https://leetcode.cn/problems/sum-of-subarray-minimums/description/
Title: 子数组的最小值之和

贡献法

参考题解：907. 子数组的最小值之和 - 力扣（Leetcode）

class Solution:
    def sumSubarrayMins(self, arr: List[int]) -> int:
        n = len(arr)

        # 左边界 left[i] 严格小于arr[i]的最近的元素位置不存在为-1
        left, st = [-1]*n, []
        for i, x in enumerate(arr):
            while st and arr[st[-1]] >= x:
                st.pop() # 移除无用数据
            if st: left[i] = st[-1]
            st.append(i)

        # 右边界 right[i] 小于等于arr[i]的最近的元素位置不存在为n(小于等于用于排除重复时多余的计算)
        right, st = [n]*n, []
        for i in range(n-1, -1, -1):#倒序遍历
            while st and arr[st[-1]] > arr[i]:
                st.pop() # 移除无用数据
            if st: right[i] = st[-1]
            st.append(i)

        ans = 0
        for i, (x, l, r) in enumerate(zip(arr, left, right)):
            ans += x * (i - l) * (r - i) #累加贡献
        return ans%