单调栈入门
题意:给你一个非负整数数组,定义某个区间的参考值为:区间所有元素的和*区间最小元素。求该数组中的最大参考值以及对应的区间。
比如说有6个数,分别为
3 1 6 4 5 2
最大参考值为6,4,5组成的区间,区间最小值为4,参考值为4*(6+5+4)=60
数据范围1<=n<=100000;
这是一道单调栈的典型例题;
就是看当当前的值入栈时,如果栈顶的值比当前的值大,则被屏蔽了后边的数,必须出栈,这里面就是在维护一个单调栈了,
没必要去关是单调递增还是单调递减。。但是怎么得到结果呢?我们知道在栈中的元素的idBIOS中间被删去的就是比栈中元素
大的,所以这样就可以利用前缀和来求当钙元素要出栈时,以该元素为最小值的区间的左右边界是什么,即L=stk[sz-1]+1;
右边界就是当前要入栈的id-1;还有一个技巧就是如果每一个值都比之前的值大,就是每个值都在栈中怎么办?这就需要最后
来一个最小的值,使得每一个值都出栈来计算了。。就是a[++n] = 0;
WA了两发,一是开始时ansl, ansr 没有初始化为1,二是 WA了 sum[]写成int了,爆范围。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string.h>
#include<algorithm>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
#include<stdlib.h>
#include<time.h>
using namespace std;
const int MAXN = 100100;
int a[MAXN],stk[MAXN];
long long sum[MAXN];
int main()
{
int i,j,n,sz = 1,ansl = 1,ansr = 1,l,r;
long long ans = 0;
cin>>n;
for(i = 1;i <= n;i++){
scanf("%d",a+i);
sum[i] += sum[i-1]+a[i];
}
a[++n] = 0;stk[0] = 0;
for(i = 1;i <= n;i++){
while(sz > 1 && a[i] < a[stk[sz-1]]){
sz--;
l = stk[sz-1];
long long tmp = 1LL*a[stk[sz]]*(sum[i-1]-sum[l]);
if(ans < tmp)
ans = tmp ,ansr = i-1,ansl = l+1;
}
stk[sz++] = i;
}
printf("%lld\n%d %d\n",ans,ansl,ansr);
}