LeetCode hot100 接雨水

题目

给定 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

提示：

• n == height.length
• 1 <= n <= 2 * 104
• 0 <= height[i] <= 105

第一遍 

第一次写hard，根本写不出来，网上看了思路，用双指针写，左边和右边分别向中间遍历，寻找最大值，装的水取决于左右两边已发现最大值的较小的一边。

class Solution {
public:
    int trap(vector<int>& height) {
        auto left_iter = height.begin();
        auto right_iter = height.end() - 1;
        int result = 0;
        int left_max = 0;
        int right_max = 0;
        while(left_iter <= right_iter){
            left_max = std::max(left_max, *left_iter);
            right_max = std::max(right_max, *right_iter);
            if(left_max < right_max){
                result += (left_max - *(left_iter));
                ++left_iter; 
            }
            else{
                result += (right_max - *(right_iter));
                --right_iter;
            }
        }
        return result;
    }
};

 

第二遍

class Solution {
public:
    int trap(vector<int>& height) {
        int left = 0;
        int right = height.size() - 1;
        int result = 0;
        int left_max = 0;
        int right_max = 0;
        while(left <= right){
            left_max = std::max(left_max, height[left]);
            right_max = std::max(right_max, height[right]);
            if(left_max < right_max){
                result += (left_max - height[left]);
                ++left; 
            }
            else{
                result += (right_max - height[right]);
                --right;
            }
        }
        return result;
    }
};