题目链接:H 小A的柱状图
又是一个新的知识点,自己慢慢积累吧,和单调栈类似的还有一个单调队列。
单调栈简单的学习:单调栈总结,不过他有一个地方写错了,在给的例子里的第(4)步,他所给的4的下标应该是4,此时栈元素因该是 2->4
这题和 HDOJ 1506 很像,不过有一个点容易搞错。
HDOJ 1506的底部宽都是一样的,在求底部长度(R[i] - L[i])的时候,发现存L[i]时多加了1,原因很好理解,可以类比:1-7之间有5个数,但是7-1=6,所以如果要求底部长度的正解,应该是 R[i]-L[i]-1(此处R和L存的是比i小的右、左端第一个元素的下标,没有进行修正)。要省去h[i] * (R[i] - L[i]-1)中的常数1,则可以在存R的时候就减一或者存L的时候就加一。
回到本题,底边长度是不一样的,第一次写的时候按自己的思维在计算地面宽度时才进行了-1操作:h[i]*(a[r[i]]-a[l[i]]-1),由于a数组存的是前缀和,所以这里相当于多算了一个矩形的宽度再-1. 正确写法应该要考虑到a[i]存的是从1-i 的宽度前缀和,即a[i]包含了第 i 个矩形,则 i 左边第一个高度小于 h[i] 的矩形不应该被算进来,需要严格减去 a[L[i]]。因此得正解:h[i]*(a[r[i]-1]-a[l[i]])。
当然,为了避免出错,还是在存R的时候就-1吧。
AC代码:
#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define pb push_back
#define Rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long ll;
typedef long double ld;
const int inf=0x3f3f3f3f;
const ll INF=9e18;
const int N=1e6+50;
ll a[N],h[N],l[N],r[N];
int top=0;
int s[N];
int main() {
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++) {
scanf("%lld",&a[i]);
a[i]+=a[i-1];
}
for(int i=1;i<=n;i++) scanf("%lld",&h[i]);
for(int i=1;i<=n;i++) {
while(top>0&&h[s[top]]>=h[i]) top--;
if(top==0) l[i]=0;
else l[i]=s[top];
s[++top]=i;
}
top=0;
for(int i=n;i>=1;i--) {
while(top>0&&h[s[top]]>=h[i]) top--;
if(top==0) r[i]=n;
else r[i]=s[top];
s[++top]=i;
}
ll ans=0;
for(int i=1;i<=n;i++)
ans=max(ans,h[i]*(a[r[i]-1]-a[l[i]]));
printf("%lld\n",ans);
return 0;
}