【PTA】 储水问题 模拟

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

数据范围：0≤n≤10⁶ ，数组中每个值满足 0 <val≤10 ⁹

输入格式:
第一行是n值，第二行是n个非负数

输出格式:
输出能接的雨水单位数

输入：

6
3 1 2 5 2 4

输出：

5

1. 数据范围1e6，所以一定要O(n)
2. 显然这是一个对最大值进行操作的题：我们要找到每一个“凹”两侧的最大值，对答案的贡献是min(最大值1，最大值2)-每一个a[i].
3. 由于要O(n)，所以只能从左到右扫一次：累加贡献；从右到左扫：减去多加的。

详看代码：(参考梦哥的代码，不愧是梦哥，太巧妙了！)

#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
typedef long long ll;
const int N=1e6+10;
int n;
int a[N];
int main()
{
	cin>>n;
	fir(i,1,n) scanf("%d",&a[i]);
	ll ans=0;
	int top=0;
	fir(i,1,n)
	{
		if(a[i]>top) top=a[i];
	}
	
	//从左往右扫，对答案的贡献是当前最大的减去当前的值 
	int now=0;//存当前最大值 
	fir(i,1,n)
	{
		if(a[i]>now) now=a[i];
		ans+=now-a[i];
	}
	
	//显然会多加
	//多加的那一部分是：左右两端的最大值，我们要按最小的算，显然上面用的是最大值的最大值
	//从右往左,减去多加的 
	now=0;
	for(int i=n;i>=1;i--)
	{
		if(a[i]==top) break;
		if(a[i]>now) now=a[i];
		ans-=top-now;//这是多加的 
	}
	cout<<ans;
	return 0;
}