[leetcode][042] Trapping Rain Water (Java)

时间:2023-03-09 05:58:13
[leetcode][042] Trapping Rain Water (Java)

我在Github上新建了一个解答Leetcode问题的Project, 大家可以参考, 目前是Java 为主,里面有leetcode上的题目,解答,还有一些基本的单元测试,方便大家起步。

题目在这里: https://leetcode.com/problems/trapping-rain-water/

[标签] Array; Stack; Two Pointers

[leetcode][042] Trapping Rain Water (Java)

这个题我感觉很难,我自己是完全不会。下面贴的是别人想到的好方法,自己适当做了些简单调整。

我觉得,这个解法非常巧妙的使用了双向的 two pointers。用到的变量有左右两个指针,一个在最左边,一个在最右边,然后两个一起往中间走。在走的时候维护两个变量,left barrier 和 right barrier,即左边的挡板和右边的挡板。

用例子说明一下思路:

用leftBarrier 和 rightBarrier围成一个”盆地“。如果是leftBarrier的高度低于rightBarrier(如下图),那么固定右边一侧,尝试移动left指针,这时我们比较leftBarrier和height[left],如果是height[left]较小的话,我们知道leftBarrier 和 left之间的部分一定是可以积攒下雨水的,(图中红色标注的部分);如果是height[left]较大的话,那就更新leftBarrier。

为什么上面是固定右边一侧的rightBarrier呢?因为当rightBarrier > leftBarrier的时候,如果再遇到height[left] < leftBarrier的情形的话,那么leftBarrier 和 height[left] 之间形成的区域是可以积攒雨水的。

啊,说的好像很绕,但是道理就是这么个道理……看看如果以后有更好的解释方法再来更新吧……汗

[leetcode][042] Trapping Rain Water (Java)

 public class Solution {
public int trap(int[] height) { int left = 0;
int right = height.length - 1; int leftBarrier = 0;
int rightBarrier = 0;
int result = 0; while (left <= right) {
if (leftBarrier <= rightBarrier) {
// there could be a basin between leftBarrier and rightBarrier
// and left side is lower one
if (height[left] > leftBarrier) {
// update left barrier
leftBarrier = height[left];
} else {
// trap water (leftBarrier - height[left]) * 1
result += leftBarrier - height[left];
}
left++;
} else {
if (height[right] > rightBarrier) {
// update right barrier
rightBarrier = height[right];
} else {
// trap water (rightBarrier - height[right]) * 1
result += rightBarrier - height[right];
}
right--;
}
}
return result;
}
}