題目

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

題解

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

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

需要注意的是出現(xiàn)相等數(shù)字時(shí)候的情況，具體表現(xiàn)為當(dāng)遇到相等數(shù)字時(shí)，不重復(fù)計(jì)算即可。可以聽左神講的，也可以自己畫一下，試一下就懂了。

class Solution {public static final int MOD = (int) (Math.pow(10, 9) + 7);public int sumSubarrayMins(int[] arr) {int L = arr.length;// 找左邊第一個(gè)【小于等于】h[i]的數(shù)// 從右向左遍歷，維護(hù)單調(diào)增棧，小/等數(shù)h[i]不斷將大數(shù)h[j]彈出，則h[i]左邊第一個(gè)小于h[i]的數(shù)為h[j]Stack<Integer> valueStack = new Stack<>();Stack<Integer> indexStack = new Stack<>();int[] leftIndex = new int[L]; // i左邊第一個(gè)小于i的數(shù)的下標(biāo)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);}// 找右邊第一個(gè)【小于】h[i]的數(shù)// 從左向右遍歷，維護(hù)單調(diào)不減棧valueStack = new Stack<>();indexStack = new Stack<>();int[] rightIndex = new int[L]; // i右邊第一個(gè)小于i的數(shù)的下標(biāo)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;} }

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。