今天继续学习单调栈解决相关问题
42.接雨水
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 1:
输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
思路:
1.看完本题后脑海中蹦出的第一个概念是木桶效应,要想知道当前列的雨水高度,我们需要当前列的高度以及其左边比他高的柱子高度和右边比他高的柱子高度,因此可以想到用单调栈来解决本题。
2.首先,单调栈内显然还是栈头到栈尾按照递增的顺序进行存储;其次,我们要想到当前元素和栈顶元素比较大小的三种情况,如果小于显然是入栈,如果相等的话就要注意,我们要先将栈顶元素出栈再将当前元素入栈(因为计算高度需要的是最右边的柱子来计算,并且单调栈存储的是下标),如果大于,那么就是我们取值的时候了。
要想算面积自然是长乘宽,宽我们可以通过当前元素的左边比他高的柱子长度和右边比他高的柱子长度取一个最小值后减去自己的高度,长度则是右边柱子下标减去左边柱子下标再减1.
class Solution {
public:
int trap(vector<int>& height) {
if(height.size() <= 2){
return 0;
}
stack<int> st;
int result = 0;
st.push(0);
for(int i = 1; i < height.size(); i++){
if(height[i] < height[st.top()]){
st.push(i);
}
else if(height[i] == height[st.top()]){
st.pop();
st.push(i);
}
else{
while(!st.empty() && height[i] > height[st.top()]){
int mid = st.top();//栈顶元素是中间矩形高度
st.pop();
if(!st.empty()){
//弹出后此时栈顶元素为左边比他高的第一个矩形高度,当前元素则是右边比他高的第一个矩形高度
int h = min(height[i], height[st.top()]) - height[mid];
int w = i - st.top() - 1;//注意减1,只要中间矩形长度
result += h * w;
}
}
st.push(i);
}
}
return result;
}
};
启发:
1.本题深刻了解到了单调栈求左边第一个更高的元素和右边第一个更高的元素的妙用,同时也将木桶效应融合在其中,本题的思路需要进一步理解。
2.后面发现实际上在相等的情况那里实际上可以不需要出栈后再入栈,直接入栈即可,因为单调栈存储的是下标,后续计算高度时首先碰到的栈顶元素一定是最右边的那一个矩形。