单调栈问题详细笔记

当元素出栈时,说明这个新元素是出栈元素向后找第一个比其小的元素

举个例子,配合下图,现在索引在 6 ,栈里是 1 5 6 。
单调栈问题详细笔记

接下来新元素是 2 ,那么 6 需要出栈。 当 6 出栈时,右边 2 代表是 6 右边第一个比 6 小的元素。

当元素出栈后,说明新栈顶元素是出栈元素向前找第一个比其小的元素

当 6 出栈时,5 成为新的栈顶,那么 5 就是 6 左边第一个比 6 小的元素。

如接雨水这题
单调栈问题详细笔记

class Solution {
    public int trap(int[] height) {
        int[] stk = new int[100010]; //递减栈,栈顶元素小
        int n = height.length;
        int tt = 0;
        int res = 0;
        for(int i = 0; i < height.length; i++){
            while(tt != 0 && height[i] > height[stk[tt]]){ //i元素严格大于栈顶元素,才出栈,相同的会入栈
                //如果进入了这层循环,则当前i元素,必定是出栈元素右侧第一个更大的数。
                int cur = stk[tt]; //当前出栈元素
                tt--;
                if(tt == 0) break;
                int bigerLeft = stk[tt]; //左侧更大元素
                res += (Math.min(height[stk[tt]], height[i]) - height[cur]) * (i - bigerLeft - 1);
            }
            stk[++tt] = i;
        }
        return res;
    }
}
上一篇:20.有效括号


下一篇:剑指Offer54.二叉搜索树的第K大节点(搜索二叉树的特点)(非递归)