【题解】
题意:给定一个n*m的矩阵,输出不被其他全为1的子矩阵完全包含的全为1的子矩阵的个数。
思路:前缀和+单调栈。对于每一列的一个位置(i,j),记录h[i][j]为该列从1到i连续的1的个数。枚举每一行作为矩阵的底边,从前往后枚举每一列,并维护一个关于h[i][j]的单调上升的栈,对于栈中的每一个值维护一个向左能拓展的位置pos。每当有元素退栈时,可以得到一个全1矩阵(i-h+1,pos)-(i,j),接下来只要判断一下能不能向下拓展就可以了。
时间复杂度O(nm)。
【代码】
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=3e3+10;
int h[maxn][maxn],stk[maxn];
int a[maxn][maxn];
char s[maxn];
ll ans;
int main()
{
int n,m; scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%s",s+1);
for(int j=1;j<=m;j++){
a[i][j]=s[j]-'0';
if(a[i][j]) h[i][j]=h[i-1][j]+1; //该列从1到i连续的1的个数
else h[i][j]=0;
}
}
for(int i=1;i<=n;i++){ //枚举每一行作为底边
int l=0,top=0;
if(a[i][1]) stk[++top]=0;
for(int j=1;j<=m;j++){ //枚举每一列
stk[++top]=j;
if(!a[i+1][j]||i==n) l=j;
int last=ans;
while(top>0&&h[i][stk[top]]>=h[i][j+1]){ //维护一个单调上升的栈
if(top>1&&stk[top-1]<l&&a[i][j]&&h[i][stk[top]]>h[i][j+1])
ans++;
top--;
}
}
}
printf("%lld\n",ans);
return 0;
}