单调栈可以求出最大矩形面积,这里稍作变形即可解决。
设s[i][j]为原图,h[i][j]为以s[i][j]为底的矩形的最大高度。记录下每个h[i][j],然后逐行求最大矩形面积即可。
不过这里最终要输出的是第二大矩形的面积(可以重叠的,比赛时想太复杂了,以为不可以重叠。。。),所以在更新的时候要有一些变化(见代码)。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<stack>
using namespace std;
typedef long long ll;
const int maxn=1e3+5;
char s[maxn][maxn];
int h[maxn][maxn];
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) scanf("%s",s[i]);
for(int j=0;j<m;++j)
{
bool start=0;
for(int i=n;i>=1;--i)
{
if(s[i][j]=='0') start=0;
else start=1;
if(start) //当前方块为1
h[i][j+1]=h[i+1][j+1]+1;
else //当前方块不为1,直接赋予0
h[i][j+1]=0;
}
}
for(int i=1;i<=n;++i) h[i][m+1]=h[i][0]=0; //这个要记得弄,这么处理会比较简单
int ans=0,ma=0; //ans记录第二大,ma记录最大
for(int i=1;i<=n;++i)
{
stack<int> S; S.push(0); //压入0,莫忘
for(int j=1;j<=m+1;++j)
{
while(h[i][j]<h[i][S.top()])
{
int id=S.top(); S.pop();
int width=j-1-S.top();
int temp=ma;
ma=max(ma,width*h[i][id]);
if(ma!=temp)
//如果ma不为原值,说明ma<width*h[i][id],那么第二大矩形的面积,就有可能是
ma原来的值,或者是最大方块的宽width-1乘高h[i][id],或者宽width乘高h[i][id]-1.
ans=max(temp,max((width-1)*h[i][id],width*(h[i][id]-1)));
//如果ma为原值,那么ans就有可能为原值,或者当前方块的面积
else ans=max(ans,width*h[i][id]);
}
S.push(j);
}
}
printf("%d\n",ans);
return 0;
}