LeetCode - Trapping Rain Water
问题描述
https://leetcode.com/problems/trapping-rain-water/
大意是给定一系列堆积方块,求当下雨的时候这些方块组成的凹槽能够容纳多少水。
解答
这道题是经同学推荐尝试的。
最早我想到的是一个比较直接的 O(n2) 的算法:从左到右每列方块进行循环(外层循环),这个需要 O(n) 的时间。每次循环里面计算“跟这列方块直接接触的水的体积”,这个计算需要 O(n) 的时间,这个过程可以通过从这列方块开始,从右往左扫描,维持一个“最高的低于该列方块”的列下标,每次刷新这个列下标的时候都可以计算出这个“直接接触的水的体积”的一部分。
不过后来觉得Hard的题应该不会那么直接,看了一眼Discuss中的高赞回答,它们都是 O(n) 的时间。其中,我发现一种解答非常巧妙:与其从一边开始扫描,不如从两边同时往中间靠。这个做法是基于一个基本观察,无论中间的方块群有多么复杂,左右最外的列可以“直接”决定左右最靠外的凹槽的水的体积。
这个算法这样用文字描述有点拗口,见代码。其中leftmax和rightmax分别代表当时最高的左右两侧对应的高度,这俩变量中的较小值可以直接反映靠近这一端凹槽的体积。于是,我们的两边的“收缩方向”必须是从两者较小的地方开始,这也是if(leftmax<=rightmax)
的意义所在。如果发现当前方块比这一边的leftmax/rightmax要矮,可以算出这一个方块上面“能存多少水”;否则,就是遇到更高的方块了,更新对应的leftmax/rightmax。
class Solution {
public:
int trap(vector<int>& height) {
int result = 0, leftmax = 0, rightmax = 0;
int left = 0, right = height.size()-1;
while(left<=right)
{
if(leftmax<=rightmax)
{
if(height[left]<=leftmax)
{
result+=leftmax-height[left];
left++;
}
else
{
leftmax=height[left];
left++;
}
}
else
{
if(height[right]<=rightmax)
{
result+=rightmax-height[right];
right--;
}
else
{
rightmax=height[right];
right--;
}
}
}
return result;
}
};