2019牛客多校第二场——H.Second Large Rectangle【单调栈/直方图】

2019牛客多校第二场——H.Second Large Rectangle【单调栈/直方图】

题目链接

题目大意：
　　给定一个 n * m 的 01 矩阵，求其中第二大的子矩阵，子矩阵元素必须全部为 1。输出其大小。
　　
思路：如果是求最大面积，可以把矩阵转化成直方图面积，单调栈乘一下，更新最大，但是这个第二大要怎莫求呢？很显然第二大必然是第一大的子矩阵，所以定义一个优先队列，每次push(h*(w-1)),push(h*w),push((h-1)*w)

#include<iostream>
#include<cstring>
#include<cstdio>
#include<stack>
#include<queue>
using namespace std;
typedef long long ll;
 
const int maxn=1000+5;
int a[maxn],l[maxn],r[maxn];
char mp[1005][1005],c;
bool vis[1005][1005];
stack<int>s;
priority_queue<int,vector<int> >q;
int n,m;
int main()
{
    q.push(0);
    q.push(0);
    int ans=0;
    scanf("%d%d",&n,&m);
    getchar();
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            scanf("%c",&c);
            mp[i][j]=c;
        }
        getchar();
    }
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            if(mp[i][j]=='1')
            {
               a[j]++;
            }
            else
            {
                a[j]=0;
            }
        }
        while(!s.empty())s.pop();
        for(int k=1; k<=m; k++)
        {
            if(k==1)
                s.push(k),l[k]=1;
            else
            {
                while(a[s.top()]>=a[k])
                {
                    s.pop();
                    if(s.empty())
                        break;
                }
                if(s.empty())
                    l[k]=1;
                else
                    l[k]=s.top()+1;
                s.push(k);
            }
            //cout<<l[k]<<"*";
        }
        while(!s.empty())s.pop();
        for(int k=m; k>=1; k--)
        {
            if(k==m)
                s.push(k),r[k]=m;
            else
            {
                while(a[s.top()]>=a[k])
                {
                    s.pop();
                    if(s.empty())
                        break;
                }
                if(s.empty())
                    r[k]=m;
                else
                    r[k]=s.top()-1;
                s.push(k);
            }
        }
        memset(vis,0,sizeof vis);
        for(int j=1; j<=m; j++)
        {
           // cout<<l[j]<<" "<<r[j]<<endl;
            if(!vis[l[j]][r[j]]&&a[j])
            {
                vis[l[j]][r[j]]=1;
                //cout<< (r[j]-l[j]+1)*a[j]<<" "<<(r[j]-l[j]+1)*(a[j]-1)<<" "<<(r[j]-l[j])*a[j]<<endl;
                q.push((r[j]-l[j]+1)*a[j]);
                q.push((r[j]-l[j]+1)*(a[j]-1));
                q.push((r[j]-l[j])*a[j]);
            }
        }
    }
    q.pop();
    cout<<q.top()<<endl;
 
}