良好的感觉（洛谷 P2422）

P2422 良好的感觉

题目描述

kkk做了一个人体感觉分析器。每一天，人都有一个感受值Ai，Ai越大，表示人感觉越舒适。在一段时间[i, j]内，人的舒适程度定义为[i, j]中最不舒服的那一天的感受值 * [i, j]中每一天感受值的和。现在给出kkk在连续N天中的感受值，请问，在哪一段时间，kkk感觉最舒适？

输入格式

第一行为N，代表数据记录的天数

第二行N个整数，代表每一天的感受值

输出格式

一行，表示在最舒适的一段时间中的感受值。

输入 #1

6
3 1 6 4 5 2

输出 #1

60

样例解释：

kkk最开心的一段时间是第3天到第5天，开心值：(6+4+5)*4=60

先从最简单的暴力思维想，我们每次枚举 i 作为最小点，然后从 i 的左边右边开始枚举，直到找到比 i
小的点，这时 i 作为区间最小点可以算出这区间的舒适程度，再更新 i ，找到最大舒适值；这个暴力想法还是比较好想，但是复杂度高； 我们可以根据这个思想，用单调栈来优化，维护一个单调递增的栈，当有元素入栈时，要出栈的元素就找到了右边比它小的元素；这个要出栈的元素左边就是比它小的元素，也就是说当元素要出栈时，左右两边比它小的元素都找到了，那么跟上面暴力的思维就差不多了，直接求和在乘就行；
代码：

#include<bits/stdc++.h>
#define LL long long 
using namespace std;
LL a[100100],sum[100100];
int sta[100100];
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
		sum[i]=a[i]+sum[i-1];//前缀和 
	}
	int ans=0;
	int l,r;
	LL S=0,M=0;
	for(int i=1;i<=n;i++){
		while(ans&&a[sta[ans]]>=a[i]){
			if(ans==1) S=(sum[i-1])*a[sta[ans]];
			else S=(sum[i-1]-sum[sta[ans-1]])*a[sta[ans]];//当要出栈时计算舒适度 
			M=max(M,S);
			ans--;//出栈 
		}
		sta[++ans]=i;//入栈 
	}
	printf("%lld\n",M);
	return 0;
}