LeetCode 42.Trapping Rain Water (接雨水)

题目描述:

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

示例:

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

Accepted C++ Solution:

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        int left=0; int right=n-1;
        int res=0;
        int maxleft=0, maxright=0;
        while(left<=right){
            if(height[left]<=height[right]){        //从低的一端注入水
                if(height[left]>=maxleft) maxleft=height[left];   //当前高度大于最大高度时,装不了水,并更新最大高度
                else res+=maxleft-height[left];
                left++;
            }
            else{                                   //从低的一端注入水
                if(height[right]>=maxright) maxright= height[right]; //同上
                else res+=maxright-height[right];
                right--;
            }
        }
        return res;
    }
};

从左到右搜索并分别保持左右最大高度,这就像容器的两边。固定较高的一端,从较低的一端注入水或者更新最大高度。

例如:如果左边的当前高度较低,我们在左边的垃圾箱中填满水。直到在中间汇拢,我们填满了整个容器。

猜你喜欢

转载自blog.csdn.net/amoscykl/article/details/82779602