一、题目介绍
接雨水是面试时经常考察的算法题,见LeetCode 42. 接雨水,题目描述如下:
该题可以用暴力解法求解,或者用双指针法优化暴力解法。这里只介绍单调栈解法。
二、单调栈解法
单调栈就是保持栈内元素有序。和【算法】代码随想录刷题记录 | 5. 栈与队列篇中的LeetCode 347. 前K个高频元素使用的优先队列一样,需要我们自己维持顺序,没有现成的容器可以用。
通常是一维数组要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时我们就要想到可以用单调栈了。
而接雨水这道题目,我们正需要寻找一个元素,右边最大元素以及左边最大元素,来计算雨水面积。
那么本题使用单调栈有如下几个问题:
1. 首先单调栈是按照行方向来计算雨水,如图:
知道这一点,后面的就可以理解了。
2. 单调栈内元素的顺序
要寻找一个元素右边第一个比自己大的元素位置,则从栈头(元素从栈头弹出)到栈底的顺序应该是从小到大的递增顺序。
因为一旦发现添加的柱子高度大于栈头元素了,此时就出现凹槽了,栈头元素就是凹槽底部的柱子,栈头第二个元素就是凹槽左边的柱子,而添加的元素就是凹槽右边的柱子。
如图:
再总结一下: 求一个元素右边第一个更大元素,单调栈就是递增的;求一个元素右边第一个更小元素,单调栈就是递减的。
3. 遇到相同高度的柱子怎么办。
推敲一下,在本题中遇到相同的元素时,新元素可以直接入栈;也可以更新栈内下标,就是将栈里元素(旧下标)弹出,将新元素(新下标)加入栈中。
4. 栈里要保存什么数值
使用单调栈,也是通过 长 * 宽 来计算雨水面积的。
长就是通过柱子的高度来计算,宽是通过柱子之间的下标来计算,那么栈里有没有必要存一个pair<int, int>类型的元素,保存柱子的高度和下标呢?
其实不用,栈里就存放下标就行,想要知道对应的高度,通过height[stack.top()] 就知道弹出的下标对应的高度了。
明确了如上几点,再来看处理逻辑。
以下逻辑主要就是三种情况
- 情况一:当前遍历的元素(柱子)高度小于栈顶元素的高度 height[i] < height[st.top()]
- 情况二:当前遍历的元素(柱子)高度等于栈顶元素的高度 height[i] == height[st.top()]
- 情况三:当前遍历的元素(柱子)高度大于栈顶元素的高度 height[i] > height[st.top()]
先将下标0的柱子加入到栈中,st.push(0);
。 栈中存放我们遍历过的元素,所以先将下标0加进来。
然后开始从下标1开始遍历所有的柱子,for (int i = 1; i < height.size(); i++)
。
如果当前遍历的元素(柱子)高度小于等于栈顶元素的高度,就把这个元素加入栈中,因为栈里本来就要保持从小到大的顺序(从栈头到栈底)。
如果当前遍历的元素(柱子)高度大于栈顶元素的高度,此时就出现凹槽了,如图所示:
取栈顶元素,将栈顶元素弹出,这个就是凹槽的底部,也就是中间位置,下标记为mid,对应的高度为height[mid](就是图中的高度1)。
此时的栈顶元素st.top(),就是凹槽的左边位置,下标为st.top(),对应的高度为height[st.top()](就是图中的高度2)。
当前遍历的元素i,就是凹槽右边的位置,下标为i,对应的高度为height[i](就是图中的高度3)。
此时大家应该可以发现其实就是栈顶和栈顶的下一个元素以及要入栈的元素,三个元素来接水!
三、题解
class Solution {
public:
int trap(vector<int>& height) {
if (height.size() == 0) return 0;
stack<int> st;
st.push(0);
int ret = 0;
for (int i = 0; i < height.size(); i++) {
if (height[i] <= height[st.top()]) {
st.push(i);
} else {
while (!st.empty() && height[i] > height[st.top()]) {
int mid = st.top();
st.pop();
if (!st.empty()) {
ret += (min(height[i], height[st.top()]) - height[mid]) * (i - st.top() - 1);
}
}
st.push(i);
}
}
return ret;
}
};