单调栈问题详细笔记

kiliwalk

关注

阅读 56

2022-02-10

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

举个例子,配合下图,现在索引在 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)

0 0 举报