LeetCode之接雨水

博主在面试字节跳动实习生岗位时，遇见了接雨水的问题。而刷过 leetcode 的同学们对此也不会陌生，因为它就是 Leetcode 上的原题——Trapping Rain Water。因此，博主在此给出解题思路和解题方案，希望能对大家有所帮助。

简介

给定 $n$ 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。例如，给定数组为 $[0,1,0,2,1,0,1,3,2,1,2,1]$，则有

上图来源于 Leetcode 的 Macros 同学。不难看出，上例的结果为6。那么，我们到底如何求解呢？

思路

经过分析和结合生活经历，我们不难发现，什么样的地形才会有积水呢？毫无疑问，自然是凹形了。你看看字，再看看图，是不是豁然开朗？那么，又如何找出凹形呢？我们不妨解一下这个字——中间低，两边高。是不是有点思路了。没错，这题的实质是寻找局部最高点。依旧以上图为例，从左边边界开始，第一个局部最高点就是第四个元素，高为2，它与左边构成了凹形，即有雨水积累。同样地，从右边边界开始，第一个局部最高点就是第十一个元素，高为2，它与右边构成了凹形，即有雨水积累。因此，我们只需要分别从左边寻找局部最高点，即可解决此问题。

代码

经过前一小节的分析，我们可有很轻松的编写出如下的代码

class Solution {
    public int trap(int[] height) {
        int h = 0;
        int area = 0;
        int i = 0, j = height.length - 1;
        while(i < j){
            if(height[i] < height[j]){
                h = Math.max(h, height[i]);
                area += h - height[i];
                i++;
            }
            else{
                h = Math.max(h, height[j]);
                area += h - height[j];
                j--;
            }
        }
        return area;
    }
}