单调栈--小A的柱状图

小A的柱状图

https://ac.nowcoder.com/acm/contest/22669/Q

链接:https://ac.nowcoder.com/acm/contest/22669/Q

来源:牛客网

柱状图是有一些宽度相等的矩形下端对齐以后横向排列的图形,但是小A的柱状图却不是一个规范的柱状图,它的每个矩形下端的宽度可以是不相同的一些整数,分别为a[i],每个矩形的高度是h[i],现在小A只想知道,在这个图形里面包含的最大矩形面积是多少。

虽然是道单调栈模板题,但是我依旧花了好久调试。对于每个柱形,我们只需要找到它能到达的最左端点和最右端点,即可计算出以该柱形为高时所组成的矩形面积。其中的宽度差值用前缀和求解即可。

代码:

#include<bits/stdc++.h>
using namespace std;
#define int long long
void solve()
{
    int n;
    cin>>n;
    int wide[n+1];
    wide[0]=0;  //为前缀和准备
    int high[n+1];
    int l[n+1];//存储每个柱能走到的最左端点
    int r[n+1];//存储每个柱能走到的最右端点
    int res=0;
    for(int i=1;i<=n;i++) {
        cin>>wide[i];
        wide[i]+=wide[i-1];
    }
    for(int i=1;i<=n;i++) cin>>high[i];
    stack<int> t;

    //寻找每个柱能走到的最左端点
    for(int i=1;i<=n;i++){
        while(!t.empty()&&high[i]<=high[t.top()]) t.pop();
        if(t.empty()) l[i]=0;  //能到达最左端
        else l[i]=t.top();
        t.push(i);
    }

    while(!t.empty()) t.pop();//清空

    //寻找每个柱能走到的最右端点
    for(int i=n;i>=1;i--){
        while(!t.empty()&&high[i]<=high[t.top()]) t.pop();
        if(t.empty()) r[i]=n;  //能到达最右端
        else r[i]=t.top()-1;//注意这个地方的索引值应该是要-1,因为这个柱子到不了栈顶端点
        t.push(i);
    }
    for(int i=1;i<=n;i++){
          res=max(res,high[i]*(wide[r[i]]-wide[l[i]]));//用前缀和求解最优矩形
    }
    cout<<res;
}
signed main()
{
    int t=1;
    //cin>>t;
    while(t--){
        solve();
    }
}
全部评论

相关推荐

人来疯的伊登等一个offer:轮抽象,还得看我这个,我甚至消息都没发😂查看图片
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务