0
点赞
收藏
分享

微信扫一扫

单调栈问题详细笔记

kiliwalk 2022-02-10 阅读 56

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

举个例子,配合下图,现在索引在 6 ,栈里是 1 5 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;
    }
}
举报

相关推荐

0 条评论