解法:单调栈
其实要接到雨水很简单,就是height[i]两边围成一个U形,肯定是一边递增一边递减。考虑这样一个思路:
索引i从左往右遍历,当height[i]递减时,索引依次入栈
- 栈底的索引对应的height肯定是当前递减序列中的最大值;
- 栈顶(刚入栈)的height肯定是当前递减序列中的最小值;
如果新来的height大于栈顶,则说明打破了递减序列,开始出现U形,此时可以计算前一个索引对应的雨水量:
class Solution {
public:
int trap(vector<int>& height) {
stack<int> downStack;
int res = 0;
for (int i(0); i < height.size(); i++){
while (!downStack.empty() && height[i] > height[downStack.top()]){
int top = downStack.top();
downStack.pop();
if (downStack.empty()) break;
int left = downStack.top();
int width = i - left - 1;
int hei = min(height[left], height[i]) - height[top];
res += width * hei;
}
downStack.push(i);
}
return res;
}
};
while循环是用于计算雨水量,条件是当前栈不为空且新的height要大于栈顶的height,才能形成U。
时间复杂度和空间复杂度均为 O(n) 。