题目:给定
n
个非负整数表示每个宽度为1
的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
思路:
维护一个单调栈,单调栈存储的是下标,满足从栈底到栈顶的下标对应的数组 height 中的元素递减。
从左到右遍历数组,遍历到下标 i 时,如果栈内至少有两个元素,记栈顶元素为 top,top 的下面一个元素是 left,则一定有 height[left]≥height[top]。如果 height[i]>height[top],则得到一个可以接雨水的区域,该区域的宽度是 i−left−1,高度是 min(height[left],height[i])−height[top],根据宽度和高度即可计算得到该区域能接的雨水量。
为了得到 left,需要将 top 出栈。在对 top 计算能接的雨水量之后,left 变成新的 top,重复上述操作,直到栈变为空,或者栈顶下标对应的 height 中的元素大于或等于 height[i]。
在对下标 i 处计算能接的雨水量之后,将 i 入栈,继续遍历后面的下标,计算能接的雨水量。遍历结束之后即可得到能接的雨水总量。
class Solution {
public int trap(int[] height) {
int ans=0;
Stack<Integer> stack=new Stack<>();
for(int i=0;i<height.length;i++){
while(!stack.isEmpty() && height[i]>height[stack.peek()]){
int top=stack.pop();
if (stack.isEmpty()) {
break;
}
int left=stack.peek();
int currWidth=i-left-1;
int currHeight=Math.min(height[left],height[i])-height[top];
ans+=currWidth*currHeight;
}
stack.push(i);
}
return ans;
}
}