Leetcode 11.42. 接雨水 C++实现

Leetcode 11. 盛最多水的容器

问题：给定一个长度为 n 的整数数组 height 。有 n 条垂线，第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。找出其中的两条线，使得它们与 x 轴共同构成的容器可以容纳最多的水。返回容器可以储存的最大水量。

说明：你不能倾斜容器。

算法：左右指针谁小谁往中间移动，遇到比之前的线矮的就跳过，再计算面积取最大。

代码：

class Solution {
public:
    int maxArea(vector<int>& height) {
        int left = 0,right = height.size() - 1;//定义左右指针
        int ans = 0;//返回值
        while(left < right){
            int area = (right - left) * min(height[left],height[right]);//面积
            ans = max(ans,area);
            if(height[left] <= height[right]){
                left++;
                while(height[left - 1] > height[left] && left < (right - 1))  left++;//优化1，如果下一条线比本条线短，就跳过不用计算。
            }
            else if(height[left] > height[right]){
                right--;
                while(height[right] < height[right + 1] && left < (right - 1))  right--;//优化2
            }
        }
        return ans;
    }
};

---

Leetcode 42. 接雨水

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

算法1 : 分别计算出每个格子的前缀最大值和后缀最大值，再减去当前格子的高 height ，就可以知道当前格子范围内能装多少水。

时间复杂度：O(n) , 空间复杂度：O(n) 

代码：

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        vector<int> pre_max(n);//前缀
        pre_max[0] = height[0];
        for(int i = 1;i < n;i++)  pre_max[i] = max(pre_max[i-1],height[i]);
        vector<int> suf_max(n);//后缀
        suf_max[n - 1] = height[n - 1];
        for(int i = n - 2;i >= 0;i--)  suf_max[i] = max(suf_max[i+1],height[i]); 
        int area = 0;
        for(int i = 0;i< n;i++)    area += min(pre_max[i],suf_max[i]) - height[i];
        return area;
    }
};

算法2 : 相向双指针，注意 while 循环可以不加等号，因为在“谁小移动谁”的规则下，相遇的位置一定是最高的柱子，这个柱子是无法接水的。

时间复杂度：O(n) , 空间复杂度：O(1) 

代码：

class Solution {
public:
    int trap(vector<int>& height) {
        int area = 0,pre_max = 0,suf_max = 0,left = 0,right = height.size() - 1;//前后缀左右指针
        while(left < right){
            pre_max = max(pre_max,height[left]);//前缀为当前从左遍历到的最大值
            suf_max = max(suf_max,height[right]);//后缀为当前从右遍历到的最大值
            area += pre_max < suf_max ? pre_max - height[left++] : suf_max - height[right--];
        }
        return area;
    }
};