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; } }
思路详解: