题目不说了,看上一篇,这一篇是用栈实现的
public static int trap(int[] height) {
int sum = 0;
if (height.length == 0) {
return 0;
}
Stack<Integer> stack = new Stack<>();
//对数组中每一个元素遍历
for (int i = 0; i <= height.length - 1; i++) {
int x = 0;
//当当前元素大于栈顶元素时取出栈顶元素计算
//如果栈顶的剩余元素的栈顶元素还是比当前元素height[i]小还要继续取然后计算
while (!stack.empty() && height[i] > height[stack.peek()]) {
//取出当前栈的栈顶元素
int last = height[stack.peek()];
//弹出元素
stack.pop();
if (stack.empty()) { // 栈空就出去
break;
}
//取出的last和左右两面墙的最低求差
x = Math.min(height[stack.peek()], height[i]) - last;
//求last和当前元素的距离用来相乘
//为什么要求x?比如第一次取栈顶是1,则两个数组距离是1,如果是第二次遍历距离就是2了
int distinct = i - stack.peek() - 1;
sum += x * distinct;
}
//将当前的下表存入栈中,注意是指针,为什么用指针,因为stack只有取栈顶的方法,而本次需要取栈第二高的数,所以存下标
stack.push(i);
}
return sum;
}