单调栈求区间长度和区间最大值

最新推荐文章于 2022-03-17 10:46:21 发布

QXK_Jack

最新推荐文章于 2022-03-17 10:46:21 发布

阅读量456

点赞数

CC 4.0 BY-SA版权

分类专栏：题解

本文链接：https://blog.csdn.net/QXK_Jack/article/details/119296197

题解专栏收录该内容

61 篇文章

订阅专栏

本文介绍了一种使用单调栈在二维矩阵中寻找由数字1组成的第二大面积的方法。通过优化算法，实现了在O(n)的时间复杂度内找到每个元素左侧或右侧最近的小于当前元素的值。文中提供了一个完整的C++实现示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

求由1组成的第二大面积

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int sum[2050][2050];
int a[2050][2050];
int n,m;
///单调栈基本应用:
///可以在O(n)下求出一个序列中
///每个元素 向右 或 向左 比它 小 或 大 的第一个元素
struct node
{
    ll pos,val;
} s[100500];
bool cmp(ll a,ll b)
{
    return a>b;

}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
            scanf("%1d",&a[i][j]);
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            if(a[i][j])
                sum[i][j]=sum[i-1][j]+a[i][j];
            else sum[i][j]=0;
        }
    }
    ll ans=0;
    ll max1=0,max2=0;
    for(int i=1; i<=n; i++)
    {
        ll top=0;
        for(int j=1; j<=m+1; j++) ///当j=m+1时，清空栈
        {
            if(top==0)
            {
                s[++top]= {j,sum[i][j]};
            }
            else
            {
                while(s[top].val>sum[i][j])
                {
                    ll temp[10]= {0};
                    temp[1]=(ll)(j-s[top-1].pos-1)*(ll)s[top].val;
                    temp[2]=(ll)(j-s[top-1].pos-2)*(ll)s[top].val;
                    temp[3]=(ll)(j-s[top-1].pos-1)*(ll)(s[top].val-1);
                    temp[4]=max1;
                    temp[5]=max2;
                    sort(temp+1,temp+1+5,cmp);
                    max1=temp[1];
                    max2=temp[2];
                    top--;
                }
                s[++top]= {j,sum[i][j]};
            }
        }
    }
    printf("%lld\n",max2);
    return 0;
}