接雨水（单调栈 STL）

hnjzsyjyj

已于 2022-03-05 18:40:43 修改

阅读量166

点赞数

分类专栏：信息学竞赛 # 栈与递归 # STL标准库文章标签：单调栈

于 2021-06-04 15:09:05 首次发布

本文链接：https://blog.csdn.net/hnjzsyjyj/article/details/117562157

版权

信息学竞赛同时被 3 个专栏收录

750 篇文章 121 订阅

订阅专栏

栈与递归

42 篇文章 1 订阅

订阅专栏

STL标准库

42 篇文章 2 订阅

订阅专栏

本文介绍如何使用动态规划算法解决一个实际问题：给定一组非负整数表示柱子高度，计算在降雨后能接收到的雨水量。通过栈操作和比较，找到最高水位并计算积水面积。实例演示了输入12个数010210132121的解题过程。

摘要由CSDN通过智能技术生成

【问题描述】
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。例如，当给定数字序列为 0,1,0,2,1,0,1,3,2,1,2,1 时，柱子高度图如下所示，最多可以接 6 个单位的雨水。

【输入格式】
第一行包含整数 n。
第二行包含 n 个非负整数。

【输出格式】
输出一个整数，表示最大接水量。

【数据范围】
1≤n≤100000,
序列中元素均不大于 1000。

【输入样例】
12
0 1 0 2 1 0 1 3 2 1 2 1

【输出样例】
6

【算法代码】

#include <bits/stdc++.h>
using namespace std;

int trap(vector<int>& h) {
	int ans=0;
	int n=h.size();

	stack<int> st;

	for(int i=0; i<n; i++) {
		if(st.empty() || h[st.top()]>h[i]) st.push(i);
		else {
			while(!st.empty() && h[st.top()]<=h[i]) {
				int idx=st.top();
				st.pop();
				if(!st.empty())
					ans+=(min(h[i],h[st.top()])-h[idx])*(i-st.top()-1);
			}
			st.push(i);
		}
	}
	return ans;
}

int main() {
	vector<int> v;
	int x;
	int n;
	cin>>n;
	for(int i=0; i<n; i++) {
		cin>>x;
		v.push_back(x);
	}

	cout<<trap(v)<<endl;

	return 0;
}

/*
in:
12
0 1 0 2 1 0 1 3 2 1 2 1
out:
6
*/

【参考文献】
https://blog.csdn.net/whutshiliu/article/details/106730218