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一秒钟。留个言点个赞呗，谢谢你#