LeetCode 面试题17.21 直方图的水量

枫流仁武

于 2021-04-02 08:39:24 发布

阅读量157

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 双指针

本文链接：https://blog.youkuaiyun.com/weixin_42379312/article/details/115390159

Leetcode 同时被 2 个专栏收录

154 篇文章

订阅专栏

双指针

11 篇文章

订阅专栏

该博客讨论了一个计算直方图能储存多少水量的问题，给出了一种解决方案。通过遍历直方图，找到最高柱子并分两部分计算可储存的水：左边柱子和右边柱子。示例中给出了输入数组和对应的输出结果6，表示可以接6个单位的水。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 1。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的直方图，在这种情况下，可以接 6 个单位的水（蓝色部分表示水）。感谢 Marcos 贡献此图。

示例:

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

class Solution {
public:
    int trap(vector<int>& height) {
        if (height.size()==0){
            return 0;
        }
        int max_idx=0;
        int max_val=0;
        for (int i = 0; i < height.size(); ++i) {
            if (height[i]>max_val){
                max_val=height[i];
                max_idx=i;
            }
        }
        int tem_max=height[0];
        int res=0;
        for (int i = 0; i < max_idx; ++i) {
            if (tem_max>height[i]) {
                res += (tem_max - height[i]);
            }
            tem_max=max(tem_max,height[i]);
        }
        tem_max=height[height.size()-1];
        for (int i = height.size()-1; i >max_idx; --i) {
            if (tem_max>height[i]){
                res+=(tem_max-height[i]);
            }
            tem_max=max(tem_max,height[i]);
        }
        return res;
    }
};