单调栈+前缀和 || Nowcoder || 牛客小白月赛13 || 小A的柱状图

题面：小A的柱状图

题解：无

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
using namespace std;
inline ll rd(){
    ll x=0;char c=getchar();
    while(c<'0'||c>'9'){c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0'; c=getchar();}
    return x;
}
const ll maxn=(1e6)+50;
ll N,A[maxn],H[maxn],sig[maxn],S[maxn],L[maxn],R[maxn],top,ans;
//前缀和维护A[i] 
//用两个单调栈，分别从左往右和从右往左找出第一个小于H[i]的数的位置
int main(){
    N=rd();
    for(int i=1;i<=N;i++){
        A[i]=rd();
        sig[i]=sig[i-1]+A[i];
    }
    for(int i=1;i<=N;i++){
        H[i]=rd();
        ans=max(ans,A[i]*H[i]);
    }
    top=1;S[top]=1;L[1]=0;
    for(int i=2;i<=N;i++){
        while(top>=1&&H[S[top]]>=H[i])top--;
        L[i]=S[top];
        S[++top]=i;
    }
    top=1;S[top]=N;R[N]=0;
    for(int i=N-1;i>=1;i--){
        while(top>=1&&H[S[top]]>=H[i])top--;
        R[i]=S[top];
        if(R[i]==0)R[i]=N+1;
        S[++top]=i;
    }
    for(int i=1;i<=N;i++)
        ans=max(ans,(sig[R[i]-1]-sig[L[i]])*H[i]);
    printf("%lld
",ans);
    return 0;
}

---

By:AlenaNuna