All-one Matrices

牛客多校第八场 A题

题目链接

单调栈+前缀和

有很多人都开了很多单调栈,其实不用,在没看别人代码之前,只看题解 我完全想不出来怎么写,看了代码才明白题解在说什么,只要一个栈就可以,我们用一个up数组代表每行第j点向上的1的个数,然后用单调栈维护大于等于当前高度的矩形,我们在用一个前缀和数组维护下一行的1,当某个高度从单调栈中被取出来时,我们用前缀和求出当前位置到取出来之后的栈顶这段的1的个数,看是否与当前位置到取出来之后的栈顶这段相等,相等的话这个矩形还能像下扩展,答案不增加,否则答案增加。

代码：

#include <bits/stdc++.h>
using namespace std;
const int  N = 3000 + 5;
int up[N], sum[N], maze[N][N];
char c[N];
int main()
{
    int n,m;
    cin >> n>> m;
    for(int i = 0 ;i < n; i++ ){
        cin >> (c + 1);
        for(int j = 1; j <= m; j ++){
            maze[i][j] = c[j] - '0';
        }
    }

    int ans = 0;
    for(int i = 0; i < n; i ++){
        stack<int > st;
        st.push(0);
        for(int j = 1; j <= m + 1;j ++){