2019牛客暑期多校训练营（第八场）A：All-one Matrices（前缀和+单调栈）

【题解】

题意：给定一个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;
}