題目

https://leetcode.com/problems/sum-of-subarray-minimums/

題解

單調棧問題。參考左神算法課：https://ke.qq.com/webcourse/3067253/103187834#taid=10646309101817205&vid=5285890813430715450

相當于找當前元素屬于哪些子序列的最小值，如下圖，以 10 位置的 7 為例，它可以是以下標 6~10 任意一個作為開始，到下標 10~14 任意一個作為結束的所有 5*5=25 個子數組的最小值，所以，以 7 為最小值的 subarray sum = 5*5*7。

需要注意的是出現相等數字時候的情況，具體表現為當遇到相等數字時，不重復計算即可。可以聽左神講的，也可以自己畫一下，試一下就懂了。

class Solution {public static final int MOD = (int) (Math.pow(10, 9) + 7);public int sumSubarrayMins(int[] arr) {int L = arr.length;// 找左邊第一個【小于等于】h[i]的數// 從右向左遍歷，維護單調增棧，小/等數h[i]不斷將大數h[j]彈出，則h[i]左邊第一個小于h[i]的數為h[j]Stack<Integer> valueStack = new Stack<>();Stack<Integer> indexStack = new Stack<>();int[] leftIndex = new int[L]; // i左邊第一個小于i的數的下標Arrays.fill(leftIndex, -1);for (int i = L - 1; i >= 0; i--) {while (!valueStack.isEmpty() && valueStack.peek() >= arr[i]) {leftIndex[indexStack.pop()] = i;valueStack.pop();}valueStack.push(arr[i]);indexStack.push(i);}// 找右邊第一個【小于】h[i]的數// 從左向右遍歷，維護單調不減棧valueStack = new Stack<>();indexStack = new Stack<>();int[] rightIndex = new int[L]; // i右邊第一個小于i的數的下標Arrays.fill(rightIndex, L);for (int i = 0; i < L; i++) {while (!valueStack.isEmpty() && valueStack.peek() > arr[i]) {rightIndex[indexStack.pop()] = i;valueStack.pop();}valueStack.push(arr[i]);indexStack.push(i);}long result = 0;for (int i = 0; i < L; i++) {int fromRange = i - leftIndex[i];int toRange = rightIndex[i] - i;result += (long)fromRange * toRange * arr[i];result %= MOD;}return (int) result;} }

總結

以上是生活随笔為你收集整理的leetcode 907. Sum of Subarray Minimums | 907. 子数组的最小值之和（单调栈）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。