Leetcode第42题(接雨水)

题目描述:

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

示例 1：输入：height = [0,1,0,2,1,0,1,3,2,1,2,1] 输出：6

解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

示例 2：输入：height = [4,2,0,3,2,5] 输出：9

class Solution {
public:
    int trap(vector<int>& height) {
        int l = 0,r = height.size() - 1;
        int lmax = 0,rmax = 0,ans = 0;
        while(l < r){
            lmax = max(lmax,height[l]);
            rmax = max(rmax,height[r]);
            if(lmax < rmax){
                ans += lmax - height[l];
                l++;
            }else{
                ans += rmax - height[r];
                r--;
            }
        }
        return ans;
    }
};

使用双指针的方式遍历数组, 并且使用lmax与rmax维护当前左边最大高度和右边最大高度, 然后将能接上的雨水累加到ans中