2019牛客暑假多校8A：All-one Matrices【单调栈】

题目：

2019牛客暑期多校8A：All-one Matrices

题意:

给定一个N*M的01矩阵，求有多少个全1子矩阵，并且这个全1子矩阵不包含在另一个更大的全1子矩阵

分析：

这题十分套路，先看最大全1子矩阵的做法，枚举每一行，将此行以上每一列的联通的1的高度求出来，那么就相当于找最大矩形面积；首先，所求的一定是每一行的每个最大矩形，假设最大矩形左右边界为【L,R】，如果下一行的【L,R】不是全1就可统计一次答案；这样做法时间O(N*N*M)，肯定是不行的；那么考虑从下到上枚举每一行，每统计一次答案，就标记【L,R】的上界，这样就可以判断当前全1子矩阵是不是包含在另一个更大的里面，注意枚举每一行的最大矩形时要去重

代码：

#include <bits/stdc++.h>

#define pii pair<int,int>
using namespace std;
const int maxn = 3e3+13;
int n,m,a[maxn][maxn],tag[maxn][maxn],pos[maxn];
char s[maxn];
int main(){
    scanf("%d %d",&n,&m);
    for(int i = 1;i <= n; ++i)
    {
        scanf("%s",s+1);
        for(int j = 1;j <= m; ++j){
            if(s[j] == '1') a[i][j] = a[i-1][j]+1;
            else a[i][j] = 0;
        }
    }
    int ans = 0;
    for(int i = n; i > 0; --i)
    {
        stack<int> s; map<pii,int> mp2; //map用来去重
        for(int j = m;j > 0; --j){
            while(!s.empty()&&a[i][j]<=a[i][s.top()]) s.pop();
            if(s.empty()) pos[j] = m+1;
            else pos[j] = s.top();
            s.push(j);
        }
        while(!s.empty()) s.pop();
        for(int j = 1;j <= m; ++j){
            int L,R = pos[j]-1;
            while(!s.empty()&&a[i][j]<=a[i][s.top()]) s.pop();
            if(s.empty()) L = 1;
            else L = s.top()+1;
            s.push(j);
            if(a[i][j] == 0) continue;
            if(!mp2.count(pii(L,R))){
                mp2[pii(L,R)] = 1;
                if(tag[L][R]==0 || tag[L][R]>i-a[i][j]+1){
                    ans++;
                    tag[L][R] = i-a[i][j]+1;
                }
            }
        }   
    }
    printf("%d",ans);
    return 0;
}