给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
输入: [0,1,0,2,1,0,1,3,2,1,2,1] 输出: 6
(1) 找到最大的位置,然后进行左右移动。
在找到最大的高度之前,设置一个左边的围栏, 左边的一定可以找到比左边围栏高的围栏,所以,只要碰到比自己低的就,记录一下水的深度,一直到比自己高的为止,然后重新记录左边的围栏。一直到最高的位置为止。
右边也是可以这样的。 一直到最高的围栏为止。
class Solution {
public:
int trap(vector<int>& height) {
long long Result=0;
int Size=height.size();
if(Size==0)
return 0;
int Begin=height[0];
int water=0;
int IndexOfMax=0;
for(int i=1;i<Size;i++){
if(height[i]>height[IndexOfMax])
IndexOfMax=i;
}
for(int i=1;i<=IndexOfMax;i++){
if(height[i]>=Begin)
{
Result +=water;
Begin =height[i];
water =0;
}else{
water +=Begin-height[i];
}
}
water=0;
Begin=height[Size-1];
for(int i=Size-2;i>=IndexOfMax;i--){
if(height[i]>=Begin)
{
Result +=water;
Begin =height[i];
water =0;
}else{
water +=Begin-height[i];
}
}
return Result;
}
};
(2)用栈去做
思路:维护一个单调栈,栈里面的元素都是严格递减的,如果当前的元素小于当前的栈顶的元素,那么久入栈,如果大于当前的栈顶的元素,那么栈顶的元素出栈,这个时候可以计算横向高度的水。一直到当前的栈顶的元素大于当前的元素。然后当前的元素入栈。
class Solution {
public:
int trap(vector<int>& height) {
int Result=0;
int SIZE=height.size();
stack<int> st;
for(int i=0;i<SIZE;i++)
{
while(!st.empty() && height[i]>=height[st.top()])
{
int buttomIndex=st.top();
st.pop();
if(st.empty())
break;
int distance=i-st.top()-1;
int waterlin=min(height[i],height[st.top()])-height[buttomIndex];
Result +=waterlin*distance;
}
st.push(i);
}
return Result;
}
};