All-one Matrices

Posted 2022-07-17 liulex

All-one Matrices

单调栈

#include<bits/stdc++.h>
#define maxn 3005
using namespace std;
#define P pair<int,int>
int n,m,ans,H[maxn][maxn],pre[maxn][maxn];
stack<P>st;
int a[maxn][maxn];
char c[maxn];//[maxn];
int main()

    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)
    
        scanf("%s",c+1);
        for(int j=1; j<=m; j++) a[i][j]=c[j]-‘0‘;
    
    for(int i=1; i<=n; i++)
    
        // tp=0;
        for(int j=1; j<=m; j++)
        
            H[i][j]=((a[i][j])?H[i-1][j]+1:0);
            pre[i][j]=pre[i][j-1]+(a[i][j]);
            //cout<<pre[i][j]<<‘ ‘;
        
        pre[i][m+1]=pre[i][m];
        //cout<<endl;
    
    int x;
    for(int i=1;i<=n;i++)
        while(!st.empty())st.pop();
        for(int j=1;j<=m+1;j++)
                x=j;
            while(!st.empty()&&st.top().first>H[i][j])
                if(i==n||(pre[i+1][j-1]-pre[i+1][st.top().second-1])<(j-st.top().second))
                    ans++;
                
                x=st.top().second;
                st.pop();
            
            if((st.empty()||H[i][j]>st.top().first))
            

                st.push(P(H[i][j],x));///以Hij为高的矩形最长能向左延伸到x
            
        
    
    printf("%d",ans);