其他分享
首页 > 其他分享> > POJ - 3494 Largest Submatrix of All 1’s(单调栈求最大全1子矩阵)

POJ - 3494 Largest Submatrix of All 1’s(单调栈求最大全1子矩阵)

作者:互联网

链接POJ - 3494 Largest Submatrix of All 1’s

题意

给出一个n×mn\times mn×m(1n,m20001\le n,m\le 20001≤n,m≤2000)的01矩阵,求最大的全为1的子矩阵,输出其面积(设矩阵单位长度为1)。



分析

在这里插入图片描述
对每一行进行考虑,均可以视为一个 柱状图,如上图,于是问题转化为为 求柱状图的最大矩形面积,这可以利用 单调栈 来处理,利用单调栈在O(m)O(m)O(m)时间内处理出 所有元素左边/右边第一个比其大/小的元素位置

在利用单调栈处理之前,还需要求出 每个位置的 “柱高”,其实只需要 对每一列进行遍历,然后 从上到下依次编号 即可。

总的时间复杂度为O(nm)O(n\cdot m)O(n⋅m)。


代码

#include<iostream>
#include<cstdio>
#include<stack>
#include<algorithm>
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int INF=0x3f3f3f3f;
const int maxn=2e3+10;
int n,m,a[maxn][maxn];
int L[maxn],R[maxn];
void prework(int i)    //预处理第i行
{
    stack<int> st;
    for(int j=1;j<=m;j++)
    {
        while(!st.empty()&&a[i][st.top()]>=a[i][j])
            st.pop();
        if(!st.empty())
            L[j]=st.top();     //a[i][j]左边第一个比其小的元素的下标
        else
            L[j]=0;
        st.push(j);
    }
    while(!st.empty())
        st.pop();
    for(int j=m;j>=1;j--)
    {
        while(!st.empty()&&a[i][st.top()]>=a[i][j])
            st.pop();
        if(!st.empty())
            R[j]=st.top();      //a[i][j]右边第一个比其小的元素的下标
        else
            R[j]=m+1;
        st.push(j);
    }
}
int main()
{
    while(~scanf("%d %d",&n,&m))
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                scanf("%d",&a[i][j]);
        for(int j=1;j<=m;j++)     //得出每个位置的”柱高”
        {
            for(int i=1,cnt=0;i<=n;i++)
            {
                if(a[i][j])
                    a[i][j]=++cnt;
                else
                    cnt=0;
            }
        }
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            prework(i);
            for(int j=1;j<=m;j++)
                ans=max(ans,a[i][j]*(R[j]-L[j]-1));
        }
        printf("%d\n",ans);
    }
    return 0;
}

墓华 发布了215 篇原创文章 · 获赞 40 · 访问量 2万+ 私信 关注

标签:int,矩阵,3494,Submatrix,st,POJ,maxn,include,empty
来源: https://blog.csdn.net/Ratina/article/details/104105417