题目链接
Leetcode.42 接雨水
hard
题目描述
给定 n n n 个非负整数表示每个宽度为 1 1 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 个单位的雨水(蓝色部分表示雨水)。
示例 2:
输入:height = [4,2,0,3,2,5]
输出:9
提示:
- n = h e i g h t . l e n g t h n = height.length n=height.length
- 1 ≤ n ≤ 2 ∗ 1 0 4 1 \leq n \leq 2 * 10^4 1≤n≤2∗104
- 0 ≤ h e i g h t [ i ] ≤ 1 0 5 0 \leq height[i] \leq 10^5 0≤height[i]≤105
解法:单调栈
维护一个 单调不增的栈 s t k stk stk,也就是 栈底元素 ≥ 栈顶元素 栈底元素 \geq 栈顶元素 栈底元素≥栈顶元素。
当遍历到 i ( 0 ≤ i < n ) i(0 \leq i < n) i(0≤i<n) 时,这时我们要讨论 h e i g h t [ i ] height[i] height[i]。
此时如果 栈 s t k stk stk不为空 并且 h e i g h t [ i ] > h e i g h t [ s t k . t o p ( ) ] height[i] > height[stk.top()] height[i]>height[stk.top()],要能够接到雨水,至少都需要 “三根柱子”( h e i g h t [ i ] = 0 height[i] = 0 height[i]=0就看成是一根高度为 0 的柱子)。
左边两根黄色的柱子在栈中。
我们把当前 i i i 记作 r r r;
弹出 此时的栈顶元素,记作 c u r cur cur;
在去当前的栈顶元素,记作 l l l;
至少有这三根柱子才能接到雨水。
雨水的高度为: h = m i n ( h e i g h t [ r ] , h e i g h t [ l ] ) − h e i g h t [ c u r ] h = min(height[r] , height[l]) - height[cur] h=min(height[r],height[l])−height[cur];
能接到的雨水总量为: t = ( r − l − 1 ) × h t = (r - l - 1) \times h t=(r−l−1)×h;
只有两根柱子,是无法接到雨水的。
时间复杂度:
O
(
n
)
O(n)
O(n)
C++代码:
class Solution {
public:
int trap(vector<int>& height) {
int n = height.size();
stack<int> stk;
int ans = 0;
for(int i = 0;i < n;i++){
while(!stk.empty() && height[i] > height[stk.top()]){
int cur = stk.top();
stk.pop();
if(stk.empty()) break;
int l = stk.top() , r = i;
int h = min(height[r] , height[l]) - height[cur];
ans += (r - l - 1) * h;
}
stk.push(i);
}
return ans;
}
};