力扣练习——15 接雨水

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

#include

#include

using namespace std;

class Solution {

public:

int trap(vector& 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;

}