LeetCode42:接雨水
给定 n 个非负整数表示每个宽度为 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 个单位的雨水(蓝色部分表示雨水)。
关键词:单调栈,栈中存放元素为数组下标,计算雨水的条件:当前遍历到的数组元素数值 > 栈顶元素所指代的数组元素数值
思路:使用单调栈存放数组下标,而且栈底到栈顶所指代的数组元素数值是依次递减的。每次当遇到:当前遍历到的数组元素的数值 > 栈顶元素所指代的数组元素的数值的时候,就计算 雨水,直到栈顶元素所指代的数组元素的数值 不大于 当前遍历到的数组元素的数值为止。然后将当前数组元素的下标入栈,继续for循环遍历下一个数组元素,直到循环结束。
代码实现:
class Solution {
public int trap(int[] height) {
if (height == null || height.length == 0) return 0;
Stack<Integer> stack = new Stack<>();
int res = 0;
for (int i = 0; i < height.length; ++i) {
while (!stack.isEmpty() && height[i] > height[stack.peek()]) {
int index = stack.pop();
if (stack.isEmpty()) {
break;
}
int width = i - stack.peek() - 1;
int length = Math.min(height[i], height[stack.peek()]) - height[index];
res += width * length;
}
stack.push(i);
}
return res;
}
}