牛客小白赛,H,单调栈模板题

题目链接: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
#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;
}

 

 

你可能感兴趣的:(牛客小白赛,H,单调栈模板题)