LeetCode(1277):统计全为 1 的正方形子矩阵 Count Square Submatrices with All Ones(Java)

2019.12.29 LeetCode 从零单刷个人笔记整理(持续更新)

github:https://github.com/ChopinXBP/LeetCode-Babel

原地动态规划。

遍历每一个位置ij时,将该位的数值置为从该位起左边的1的总个数,例如对于矩阵

  [0,1,1,1],
  [1,1,1,1],
  [0,1,1,1]

遍历结束后效果为:

  [0,1,2,3],
  [1,2,3,4],
  [0,1,2,3]

为了统计正方形矩阵的个数,可以假定当前位置ij为正方形的右下角点,其值为可能的最大边长maxlen = matrix[i][j]。

沿着同一列j向上遍历,遍历行数即为当前正方形的边长curlen,当遍历过程的最小边长minlen >= curlen时,可以构成当前正方形。


传送门:统计全为 1 的正方形子矩阵

Given a m * n matrix of ones and zeros, return how many square submatrices have all ones.

给你一个 m * n 的矩阵,矩阵中的元素不是 0 就是 1,请你统计并返回其中完全由 1 组成的 正方形 子矩阵的个数。

示例 1:
输入:matrix =
[
  [0,1,1,1],
  [1,1,1,1],
  [0,1,1,1]
]
输出:15
解释: 
边长为 1 的正方形有 10 个。
边长为 2 的正方形有 4 个。
边长为 3 的正方形有 1 个。
正方形的总数 = 10 + 4 + 1 = 15.

示例 2:
输入:matrix = 
[
  [1,0,1],
  [1,1,0],
  [1,1,0]
]
输出:7
解释:
边长为 1 的正方形有 6 个。 
边长为 2 的正方形有 1 个。
正方形的总数 = 6 + 1 = 7.


/**
 *
 * Given a m * n matrix of ones and zeros, return how many square submatrices have all ones.
 * 给你一个 m * n 的矩阵,矩阵中的元素不是 0 就是 1,请你统计并返回其中完全由 1 组成的 正方形 子矩阵的个数。
 *
 */

public class CountSquareSubmatricesWithAllOnes {
    public int countSquares(int[][] matrix) {
        int result = 0;
        for(int i = 0; i < matrix.length; i++) {
            for(int j = 0; j < matrix[0].length; j++) {
                if(matrix[i][j] != 1){
                    continue;
                }
                result++;
                if(j == 0){
                    matrix[i][j] = 1;
                    continue;
                }
                matrix[i][j] = matrix[i][j - 1] + 1;
                int maxlen = matrix[i][j];
                int minlen = maxlen;
                int row = i - 1;
                int curlen = 2;
                while(row >= 0 && curlen <= maxlen){
                    minlen = Math.min(matrix[row][j], minlen);
                    if(minlen < curlen){
                        break;
                    }
                    result++;
                    row--;
                    curlen++;
                }
            }
        }
        return result;
    }
}




#Coding一小时,Copying一秒钟。留个言点个赞呗,谢谢你#

你可能感兴趣的:(数据结构与算法,JAVA,LeetCode)