题目描述
原题链接:42. 接雨水
解题思路
(1)暴力法
每遍历到一个高度时,就左右寻找探索两侧的边界,然后取二者中的最小值与当前高度相减(取最小值的原因,相当于是一个最短的木桶板决定了它最大可存水量的上界)。
class Solution {
public:
int trap(vector<int>& height) {
int n = height.size(), res = 0;
for(int i = 0; i < n; i++) {
// 遇到第一个和最后一个高度不左右探索
if(i == 0 || i == n - 1) continue;
int left = height[i], right = height[i];
// 寻找左边界
for(int l = i - 1; l >= 0; l--) {
if(height[l] > left) left = height[l];
}
// 寻找右边界
for(int r = i + 1; r < n; r++) {
if(height[r] > right) right = height[r];
}
// 找到可存水量之差
int h = min(left, right) - height[i];
if(h > 0) res += h;
}
return res;
}
};
此方式会超时。
Python
class Solution:
def trap(self, height: List[int]) -> int:
n = len(height)
res = 0
for i in range(n):
if i == 0 or i == n - 1:
continue
left = right = height[i]
for j in range(i - 1, -1, -1):
if height[j] > left:
left = height[j]
for j in range(i + 1, n):
if height[j] > right:
right = height[j]
h = min(left, right) - height[i]
res += h
return res
(2)单调栈
单调栈主要用于解决找到某一侧遇到的第一个更大或更小的值,因此该题可用单调栈来存储来”未达到封口凹槽“的一边。
存储元素:数组下标,存储次序:从栈底到栈顶,单调不增(包含递减和等于的情况)。
由于目标是让求出接多少雨水,也就相当于是求可接雨水体积。用单调栈来存储高度值,当为单调不增时入栈,当遇到的数比栈顶元素大时,就把栈顶元素弹出,计算当前栈顶元素可接的雨水体积。由于有可能存在,当前元素比栈顶元素大,但比栈顶元素的左元素小的情况,也有可能当前元素比栈顶元素的左元素大。然后,可接的个数应该取“最短的那个木板”。因此,寻找二者中的最小高度作为与栈顶元素的高度差,相减求出可接雨水高度。然后,再找出长度与之相乘,就可得到可接的雨水体积,也就是可接雨水个数。
class Solution {
public:
int trap(vector<int>& height) {
int n = height.size(), res = 0;
stack<int> st;
st.push(0);
for(int i = 1; i < n; i++) {
while(!st.empty() && height[st.top()] < height[i]) {
int mid = st.top(); st.pop();
if(!st.empty()) {
// 找到高度边界中的最小值(最短的那块木板) - 本次高度
int h = min(height[st.top()], height[i]) - height[mid];
// 右凹槽下标 - 左凹槽下标 - 1 (因为只需要到中间凹槽,会少减去一个1)
int w = i - st.top() - 1;
res += h * w;
}
}
// 当为单调不增时或弹栈后都入栈当当前情况,用于后续对比
st.push(i);
}
return res;
}
};
Python
class Solution:
def trap(self, height: List[int]) -> int:
n = len(height)
stack = []
res = 0
for i in range(n):
while stack and height[stack[-1]] < height[i]:
mid = height[stack[-1]]
stack.pop()
if stack:
h = min(height[stack[-1]], height[i]) - mid
w = i - stack[-1] - 1
res += h * w
stack.append(i)
return res
参考文章:42. 接雨水