题目大意:
给出一个n*m的01矩阵,问有多少个面积大等于k的全0子矩阵。n,m<=4000。
解题思路:
开始时以为用单调栈处理出以每个点为右下角的极大子矩阵算包含右下角的合法矩阵就行了,但每个点会产生O(n)个极大子矩阵,且会算重。
后来看了题解,可以加入一个高度时不计算贡献,在弹掉该高度时计算以它为高度的极大子矩阵中高度大于周围两个高度的部分的贡献,这样就是
O(n2)
O
(
n
2
)
的了。
说起来有点绕,还是看代码吧。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
int i=0,f=1;char c;
for(c=getchar();c!='-'&&(c<'0'||c>'9');c=getchar());
if(c=='-')f=-1,c=getchar();
for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
return i*f;
}
const int N=4005;
int n,m,k,a[N][N],f[N][N],s[N],h[N];
ll ans;
int main()
{
//freopen("lx.in","r",stdin);
n=getint(),m=getint(),k=getint();
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
f[i][j]=f[i-1][j]+f[i][j-1]-f[i-1][j-1]+(i*j>=k);
a[i][j]=getint();
}
for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)f[i][j]+=f[i][j-1];
for(int i=1,top;i<=n;i++)
{
top=0;
for(int j=1;j<=m;j++)h[j]=a[i][j]?0:++h[j];
for(int j=1;j<=m+1;j++)
{
while(top&&h[s[top]]>=h[j])ans+=f[h[s[top]]][j-1-s[top-1]]-f[max(h[s[top-1]],h[j])][j-1-s[top-1]],top--;
s[++top]=j;
}
}
cout<<ans<<'\n';
return 0;
}