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;
}