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;
}