力扣练习——15 接雨水

1.问题描述

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

 

 

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 

 

示例:

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

输出: 6

 

可使用以下代码，完成其中的trap函数，其中形参height为上述的柱子高度数组，返回接到的雨水量。

#include<iostream>

#include<stack>

#include<vector>

using namespace std;

 

class Solution {

public:

int trap(vector<int>& height)

{

             //填充本函数完成功能  

}

};

int main()

{

    int h;

    vector<int> v;

    while(cin>>h)

    {

        v.push_back(h);

    }

    int res=Solution().trap(v);

    cout<<res<<endl;

    return 0;

}

 

2.输入说明

输入若干个非负整数，以空格分隔。

3.输出说明

输出一个整数，表示结果。

4.范例

输入

0 1 0 2 1 0 1 3 2 1 2 1

输出

6

5.代码

#include<iostream>
#include<algorithm>
#include<stack>
#include<vector>

using namespace std;



class Solution {

public:

    int trap(vector<int>& height)
        //思想：维护一个单调栈，单调栈存储的是下标，满足从栈底到栈顶的下标对应的数组 \textit{height}height 中的元素递减。
        //题解https://leetcode.cn/problems/trapping-rain-water/solution/jie-yu-shui-by-leetcode-solution-tuvc/
    {
        //1.定义初始值
        int ans = 0;
        stack<int>s;//定义栈
        int len = height.size();
        //2.遍历并计算接雨量
        for (int i = 0; i < len; i++)
        {
            while (!s.empty() && height[i] > height[s.top()])//计算i处的接雨量
            {
                int top = s.top();
                s.pop();//出栈
                if (s.empty())//栈空了，直接退出循环
                    break;

                int left = s.top();//left变成新的top
                int currWidth = i - left - 1;//新的宽度
                int currHeight = min(height[left], height[i]) - height[top];//新的高度

                ans += currWidth * currHeight;
            }//while
            s.push(i);//在对下标 i 处计算能接的雨水量之后，将 i入栈，继续遍历后面的下标，计算能接的雨水量。
        }
        return ans;
    }

};

int main()

{

    int h;

    vector<int> v;

    while (cin >> h)

    {

        v.push_back(h);

    }

    int res = Solution().trap(v);

    cout << res << endl;

    return 0;

}