907. 子数组的最小值之和

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 * 104
  • 1 <= arr[i] <= 3 * 104

动态规划

class Solution {
    public int sumSubarrayMins(int[] arr) {
        int n = arr.length;
        long ans = 0;
        final int MOD = 1000000007;
        Deque<Integer> monoStack = new ArrayDeque<Integer>();
        int[] dp = new int[n];

        for (int i = 0; i < n; i++) {
            // 栈非空   且   栈顶元素大于数组第i个数, 出栈
            while (!monoStack.isEmpty() && arr[monoStack.peek()] > arr[i]) {
                monoStack.pop();
            }
            // 栈为空时,k = i + 1;否则k = i - 栈顶元素
            
            int k = monoStack.isEmpty() ? (i + 1) : (i - monoStack.peek());
            // 存储每个元素为结尾的子序列最小值之和
            dp[i] = k * arr[i] + (monoStack.isEmpty() ? 0 : dp[i - k]);
            ans = (ans + dp[i]) % MOD;
            monoStack.push(i);
        }

        return (int) ans;
    }
}

 思路详解:

 

 



posted on 2022-10-28 11:21  HHHuskie  阅读(27)  评论(0编辑  收藏  举报

导航