2019牛客暑期多校训练营（第八场 ) A All-one Matrices （单调栈）

题目链接：https://ac.nowcoder.com/acm/contest/888/A

题意：给你一个0,1矩阵，要求找出所有  不可扩大的全1矩阵，就是所有的全1矩阵不能有包含关系

思路：先预处理每个点往上1的最大高度，然后再维护  每一行的前缀和sum[i][j]， 然后用单调栈求出以h[i][j] 为矩阵高时的右边界R，和左边界L，然后判断一下此矩阵可不可扩展，用  sum[i+1][R] - sum[i+1][L-1] == R - L + 1  来判断，每一行可能求出多个相同矩阵，所以需要去重，用单调栈维护一个增序列

 

 

#include<bits/stdc++.h>
using namespace std;
const int N = 3e3+5;
int n,m,h[N][N],sum[N][N],q[N],R[N],L[N];
char s[N][N];
int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) {
        scanf("%s", s[i] + 1);
        for (int j = 1; j <= m; j++) {
            if (s[i][j] == '1')
                h[i][j] = h[i - 1][j] + 1;
            sum[i][j] = sum[i][j - 1] + s[i][j] - '0';
        }
    }
    int ans = 0;
    for (int i = n; i; i--) {
        stack<int> sta;
        sta.push(m + 1);
        ///从后往前找出每个点j以h[i][j]（1的高度）为高度的矩形右边界R[j]
        for (int j = m; j; j--) {
            while (!sta.empty() && h[i][j] <= h[i][sta.top()])
                sta.pop();
            if (sta.empty()) R[j] = -1;
            else R[j] = sta.top() - 1;
            sta.push(j);
        }
        while (!sta.empty()) sta.pop();
        sta.push(0);
        ///从前往后找出每个点j以h[i][j]（1的高度）为高度的矩形左边界L[j]
        for (int j = 1; j <= m; j++) {
            while (!sta.empty() && h[i][j] <= h[i][sta.top()]) sta.pop();
            if (sta.empty()) L[j] = -1;
            else L[j] = sta.top() + 1;
            sta.push(j);
        }
        while (!sta.empty()) sta.pop();
        for (int j = 1; j <= m; j++) {
            if (h[i][j] == 0) {
                while (!sta.empty())
                    sta.pop();
                continue;
            }
            while (!sta.empty() && h[i][j] < h[i][sta.top()]) sta.pop();
            if (sta.empty() || h[i][j] != h[i][sta.top()]) {///去重
                int l = L[j];
                int r = R[j];
                if (sum[i + 1][r] - sum[i + 1][l - 1] != r - l + 1)///确定下边界(下面一排不全是1，不会被覆盖)
                    ans++;
                sta.push(j);
            }
        }
    }
    printf("%d\n", ans);
    return 0;
}