42.接雨水

解法:单调栈

其实要接到雨水很简单,就是height[i]两边围成一个U形,肯定是一边递增一边递减。考虑这样一个思路:

索引i从左往右遍历,当height[i]递减时,索引依次入栈

  • 栈底的索引对应的height肯定是当前递减序列中的最大值;
  • 栈顶(刚入栈)的height肯定是当前递减序列中的最小值;

如果新来的height大于栈顶,则说明打破了递减序列,开始出现U形,此时可以计算前一个索引对应的雨水量:

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> downStack;
        int res = 0;
        for (int i(0); i < height.size(); i++){
            while (!downStack.empty() && height[i] > height[downStack.top()]){
                int top = downStack.top();
                downStack.pop();
                if (downStack.empty()) break;
                int left = downStack.top();
                int width = i - left - 1;
                int hei = min(height[left], height[i]) - height[top];
                res += width * hei;
            }
            downStack.push(i);
        }
        return res;
    }
};

while循环是用于计算雨水量,条件是当前栈不为空且新的height要大于栈顶的height,才能形成U。

时间复杂度和空间复杂度均为 O(n) 。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值