C++直方图水量(接雨水问题)

在这里插入图片描述
第一种方法:双指针
利用左右指针的下标差值计算出每一层雨水和柱子的体积。如下图,第一层体积为11,第二层为8,第三层为1。累加得到整体体积area= 20(注意:每一层从左边第一个方格到右边最后一个方格之间一定是被蓝黑两种颜色的方格填满,不会存在空白,所以我们可以这么求值)
计算柱子的总体积Sum,就是height:[0,1,0,2,1,0,1,3,2,1,2,1]数组之和14
返回结果area− Sum就是雨水的体积。
在这里插入图片描述

class Solution {
public:
    int trap(vector<int>& height) {
        int sum = accumulate(height.begin(), height.end(), 0);
        int area = 0;
        int floor = 1, left = 0, right = height.size() - 1;
        while (left <= right) {
            while (left <= right && height[left] < floor) {
                left++;
            }
            while (left <= right && height[right] < floor) {
                right--;
            }

            area += (right - left + 1);
            floor += 1;

        }

        return area - sum;
    }
};

方法二:单调栈
除了计算并存储每个位置两边的最大高度以外,也可以用单调栈计算能接的水的总量。

维护一个单调栈,单调栈存储的是下标,满足从栈底到栈顶的下标对应的数组 height 中的元素递减。

从左到右遍历数组,遍历到下标 ii 时,如果栈内至少有两个元素,记栈顶元素为 top,top 的下面一个元素是 left,则一定有 height[left]≥height[top]。如果 height[i]>height[top],则得到一个可以接雨水的区域,该区域的宽度是i−left−1,高度是min(height[left],height[i])−height[top],根据宽度和高度即可计算得到该区域能接的水的量。
为了得到left,需要将top 出栈。在对 top 计算能接的水的量之后,left 变成新的 top,重复上述操作,直到栈变为空,或者栈顶下标对应的height 中的元素大于或等于height[i]。
在对下标 i 处计算能接的水的量之后,将 i 入栈,继续遍历后面的下标,计算能接的水的量。遍历结束之后即可得到能接的水的总量。

class Solution {
public:
    int trap(vector<int>& height) {
        int ans = 0;
        stack<int> stk;
        int n = height.size();
        for (int i = 0; i < n; ++i) {
            while (!stk.empty() && height[i] > height[stk.top()]) {
                int top = stk.top();//上一个left位置,栈下面的值也一定比top大,且位置在top前,right>left,中间就能形成接水区域
                stk.pop();
                if (stk.empty()) {
                    break;
                }
                int left = stk.top();//left后移,进行计算
                int currWidth = i - left - 1;
                int currHeight = min(height[left], height[i]) - height[top];//一层一层计算面积
                ans += currWidth * currHeight;
            }
            stk.push(i);
        }
        return ans;
    }
};
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值