leetcode Trapping Rain Water

AC代码
时间复杂度为O(n),空间复杂度为O(1)。只用了i,j,k三个变量作为指针。
在这里插入图片描述
思路是:设置两个指针i,j,i指向左边的第一个高大于0的柱子,j指向右边第一个高大于0的柱子。然后,选择这两课柱子中较矮的那个(若两者相等,则选择i),从这个位置开始直到其后(或者其前)的第一个大于它的高度的柱子之间,统计这段区间内的面积。
如这里第一次循环时选择i,则i向右移动到第二个黑色柱子处停下(此时i=3),统计从i=1到i=3之间的距离为1。
第一次循环开始时,i,j的位置,以及第一次循环结束时i的位置i'
上图所示是第一次循环开始时,i,j的位置,以及第一次循环结束时i的位置i’
第二次循环时选择j,j向左移动到第二个黑色柱子处并统计j和j‘之间的面积(为0)。

class Solution {
public:
    int trap(vector<int>& height) {
        int i = 0,j = height.size()-1;
        while(i<height.size() && height[i]==0){
            i++;
        }
        while(j>=0 && height[j] ==0){
            j--;
        }
        int area = 0;
        while(i<j){
            if(height[i]<=height[j]){
                int k = i + 1;
                while(k<j && height[k]<=height[i]){
                    area += height[i] - height[k];
                    k++;
                }
                i = k;
            }else{
                int k = j - 1;
                while(i<k && height[k]<height[j]){
                    area += height[j] - height[k];
                    k--;
                }
                j = k;
            }
        }
        return area;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值