PTA储水问题(单调栈)

给定一系列非负整数，将这些数据看作一个柱子高度图，计算按此排列的柱子，下雨之后能接多少雨水。(数组以外的区域高度视为0)

数据范围：0≤n≤106
，数组中每个值满足 0 <val≤109

输入格式:

第一行是n值，第二行是n个非负数

输出格式:

输出能接的雨水单位数

输入样例:

6
3 1 2 5 2 4

输出样例:

说明：柱子高度分别为3,1,2,5,2,4, 如图所示。在这种情况下，可以接 5个单位的雨水，蓝色的为雨水。

解法：使用单调栈(单调递增栈：单调递增栈就是从栈底到栈顶数据是从大到小)。

若栈为空或当前读入元素x小于等于栈顶，x的下标入栈；

若当前读入的元素x大于栈顶下标对应值，出栈给t，此时产生的储水区域为：

以a[t]为水平线，t到栈顶下标为长度，min(x,a[栈顶])为高度构成的(矩形)区域

AC代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int a[1000000];
stack<int>s;
int main()
{
	int n;
	cin>>n;
	ll res=0;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
		while(!s.empty()&&a[i]>a[s.top()])
		{
			int t=s.top();
			s.pop();
			if(s.empty()) 
			    break;
			ll h=min(a[i],a[s.top()]);
			res+=(i-s.top()-1)*(h-a[t]);
		}
		s.push(i);
	}
	cout<<res;
	return 0;
}