leetcode 1504.统计全1子矩形

原题

1504.统计全1子矩形
2020年7月5日 196周赛第三题

题解

方法一

暴力法。首先四个循环确定矩形的长方形的左上和右下角，之后逐个比对矩形里面的数字是不是1。
本思路java代码示例：

/*暴力法通过
@v7fgg
执行用时：285 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗：40.9 MB, 在所有 Java 提交中击败了100.00%的用户
2020年7月5日 13:13
*/
class Solution {
    public int numSubmat(int[][] mat) {
        int ans=0;
        int m=mat.length;
        int n=mat[0].length;
        //ij确定左上角，kl确定右下角
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                for(int k=i;k<m;k++){
                    for(int l=j;l<n;l++){
                        //逐个检查
                        boolean bukeyi=false;//表示此矩形是不是全1
                        for(int a=i;a<=k;a++){
                            for(int b=j;b<=l;b++){
                                if(mat[a][b]==0){
                                    bukeyi=true;
                                    break;
                                }
                            }
                            if(bukeyi){break;}                   
                        }
                        if(bukeyi){break;}
                        ans++;                    
                    }
                }
            }
        }
        return ans;
    }
}

方法二 逐行搜索连续1

先设立一个矩阵maxZ，其中maxZ(i,j)表示在mat的第i行第j列时，这一行到这里已经有多少个连续的1了。之后再遍历一次maxZ，定位每个点maxZ(i,j)，并且每次都以这个点为右下角，向上扩展一行，并计算出扩展行后有多少全1矩形，并加到计数中。
本思路java代码示例：

/*
@v7fgg
执行用时：13 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗：40.7 MB, 在所有 Java 提交中击败了100.00%的用户
2020年7月5日 14:22
*/
class Solution {
    public int numSubmat(int[][] mat) {
        int ans=0;
        int m=mat.length;
        int n=mat[0].length;
        int maxZ[][]=new int[m][n];
        for(int i=0;i<m;i++){
            int maxZuo=0;
            for(int j=0;j<n;j++){
                if(mat[i][j]==1){
                    maxZuo++;
                }
                else{maxZuo=0;}
                maxZ[i][j]=maxZuo;
            }
        }
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                int rec=150;
                for(int k=i;k>=0;k--){
                    rec=Math.min(rec,maxZ[k][j]);
                    ans+=rec;
                }
            }
        }
        return ans;
    }
}