Leecode 42题“接雨水”问题有多个解法,其中又以双指针解法最为简洁。但是官方题解似乎没有过多的解释双指针解法的正确性,而精选题解的解释又可能会产生歧义。下面简单分享一下我的理解,如有不对欢迎指正。
官方题解
我们可以认为如果一端有更高的条形块(例如右端),积水的高度依赖于当前方向的高度(从左到右)。当我们发现另一侧(右侧)的条形块高度不是最高的,我们则开始从相反的方向遍历(从右到左)。
我们必须在遍历时维护 left_max 和right_max ,但是我们现在可以使用两个指针交替进行,实现 1 次遍历即可完成。
public int trap(int[] height) {
int left = 0, right = height.length - 1;
int ans = 0;
int left_max = 0, right_max = 0;
while (left < right) {
if (height[left] < height[right]) {
if (height[left] >= left_max) {
left_max = height[left]; //更新 left_max
} else {
ans += (left_max - height[left]);
}
++left;
} else {
if (height[right] >= right_max) {
right_max = height[right]; //更新 right_max
} else {
ans += (right_max - height[right]);
}
--right;
}
}
return ans;
}
作者:LeetCode
链接:https://leetcode-cn.com/problems/trapping-rain-water/solution/jie-yu-shui-by-leetcode/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
我的一点理解
在算法运行过程中,left 指向下标 i ,假设 left 左边的最大高度出现在下标 x,即 height [x] = left_max。现在让时光倒流,我们想象一下之前 left 指针刚到 x 的时候,必定是因为 height [x] < height [y] (y是当时 right 指针指向的下标),所以 left 指针才能划过 x 并更新 max_left。也就是说,right 的右边(包括 right 本身指向的地方)肯定存在一根柱子的高度大于 left_max,进一步推理得到,left 右边肯定存在一根柱子高于 left_max。
现在回到原来的运行状态,判断 height[left] < height[right] 是否成立:
-
如若成立,说明不但 max_left 不如 right 右边的某根柱子高,而且 left 指向的柱子也不如 right
指向的柱子高;所以 left 左边的最大高度小于 left 右边的最大高度,同时 left 本身的高度也小于 left 右边的最大高度,此时便可以计算 left 的积水量和更新 max_left。 -
如果 height[left] >= height[right] ,思路跟上面的思路是对称的,把 left 和 right 交换过来想就好了。
如有不对之处,欢迎批评指正