CSU 1838 Water Pump(单调栈)

Water Pump

【题目类型】单调栈

&题解：

这题可以枚举缺口,共n-1个,之后把前缀面积和后缀面积用O(n)打一下表,最后总面积减去前缀的i个和后缀的n-1-i个面积就是现在第i+1给缺口漏水的面积

&代码：

#include <cstdio>
#include <iostream>
#include <set>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <map>
#include <stack>
#include <queue>
#include <vector>
using namespace std;
#define INF 0x3f3f3f3f3f3f3f3f
using ll=long long;
const int maxn= 1e5 +9;
int n,a[maxn];
ll le[maxn],ri[maxn];
stack<pair<int,int> > sti;
ll sol(int t[],ll an[])
{
	sti.emplace(t[0],0);
	for(int i=1;i<n;i++){
		while(sti.size()>1&&sti.top().first<=t[i]) sti.pop();
		int fi=sti.top().first,se=sti.top().second;
		if(sti.size()==1){
			if(t[i]<fi){
				an[i]=min(fi,t[i])*(i-se)+an[se];
			}
			else{
				an[i]=min(fi,t[i])*(i-se)+an[se];
				sti.pop();
			}
		}
		else{
			an[i]=min(fi,t[i])*(i-se)+an[se];
		}
		sti.emplace(t[i],i);
	}
}
int main()
{
//	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	freopen("E:1.txt","r",stdin);
	int T;scanf("%d",&T);
	while(T--){
		scanf("%d",&n);
		for(int i=0;i<n;i++)
			scanf("%d",a+i);
		while(sti.size()) sti.pop();
		sol(a,le);
		reverse(a,a+n);
		while(sti.size()) sti.pop();
		sol(a,ri);
		reverse(ri,ri+n);
//		for(int i=1;i<n;i++){
//			cout<<le[i]<<" ";
//		}cout<<endl;
//		for(int i=1;i<n;i++){
//			cout<<ri[i]<<" ";
//		}cout<<endl;
		ll ans=0;
		for(int i=1;i<n;i++){
//			printf("%lld====
",le[n-1]-le[i]-ri[i+1]);
			ans=max(ans,abs(le[n-1]-(le[i]+ri[i+1])));
		}
		printf("%lld
",ans);
		memset(a,0,sizeof(a));
		for(int i=0;i<n;i++){
			le[i]=ri[i]=0;
		}
	}
	return 0;
}