单调栈问题详细笔记
作者:互联网
当元素出栈时,说明这个新元素是出栈元素向后找第一个比其小的元素
举个例子,配合下图,现在索引在 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;
}
}
标签:出栈,int,tt,元素,笔记,height,详细,stk,单调 来源: https://blog.csdn.net/HC_wood/article/details/122856951