题目:
给定 n
个非负整数表示每个宽度为 1
的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
方法一(相向双指针):竖着计算面积
代码:
class Solution {
public int trap(int[] height) {
int ans = 0, left = 0, right = height.length - 1, preMax = 0, sufMax = 0;
while (left < right) {
preMax = Math.max(preMax, height[left]);
sufMax = Math.max(sufMax, height[right]);
ans += preMax < sufMax ? preMax - height[left++] : sufMax - height[right--];
}
return ans;
}
}
方法二(单调栈):横着计算面积
代码:
class Solution {
public int trap(int[] height) {
int ans = 0;
Deque<Integer> st = new ArrayDeque<>();
for (int i = 0; i < height.length; i++) {
while (!st.isEmpty() && height[i] >= height[st.peek()]) {
int bottomH = height[st.pop()];
if (st.isEmpty()) {
break;
}
int left = st.peek();
int dh = Math.min(height[left], height[i]) - bottomH; // 面积的高
ans += dh * (i - left - 1);
}
st.push(i);
}
return ans;
}
}