思路清奇,代码简洁的好题
问题大体分两部:
- 记录子矩阵个数,统计每一个点作为右下角时可以得到多少矩形,加起来就是答案
- 剪掉墙挡住的地方
考虑从右下角开始,如果有0已经挡住了矩阵,那么更靠左、上的0都卵用没有->单调栈
#include<bits/stdc++.h>
using namespace std;
int n,m,a[3005][3005];
int up[3005],s[3005],top;
long long sum[3005],ans=0;
int main()
{
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 i=1;i<=n;i++)
{
top=0;//维护最近的0
for(int j=1;j<=m;j++)
{
if(!a[i][j]) up[j]=i;//墙
while(top&&up[s[top]]<up[j]) top--; //远的0不要
s[++top]=j;//高度
sum[top]=sum[top-1]+(i-up[s[top]])*(s[top]-s[top-1]);//左边的答案继承+新形成的矩形方案(新的最大矩形长 ×宽)
ans+=sum[top];
}
}
printf("%lld",ans);
return 0;
}