Leetcode 1277 统计全为1的正方形子矩阵

Leetcode 1277 统计全为1的正方形子矩阵

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

思路：动态规划
递推公式推导如下：（参考LeetCode官方题解）
用 f[i][j] 表示以 (i, j) 为右下角的正方形的最大边长，那么除此定义之外，f[i][j] = x 也表示以 (i, j) 为右下角的正方形的数目为 x

• f[i][j] 与相邻位置的关系
  
  如上图所示，若对于位置 (i, j) 有 f[i][j] = 4，我们将以 (i, j) 为右下角、边长为 4 的正方形涂上色，可以发现其左侧位置 (i, j - 1)，上方位置 (i - 1, j) 和左上位置 (i - 1, j - 1) 均可以作为一个边长为 4 - 1 = 3 的正方形的右下角。也就是说，这些位置的的 f 值至少为 3，即：

f[i][j - 1] >= f[i][j] - 1
f[i - 1][j] >= f[i][j] - 1
f[i - 1][j - 1] >= f[i][j] - 1

将这三个不等式联立，可以得到：

min(f[i][j-1],f[i-1][j],f[i-1][j-1])>=f[i][j]-1

• 同理，我们也可以固定 f[i][j] 相邻位置的值，得到另外的限制条件。

如上图所示，假设 f[i][j - 1]，f[i - 1][j] 和 f[i - 1][j - 1] 中的最小值为 3，也就是说，(i, j - 1)，(i - 1, j) 和 (i - 1, j - 1) 均可以作为一个边长为 3 的正方形的右下角。我们将这些边长为 3 的正方形依次涂上色，可以发现，如果位置 (i, j) 的元素为 1，那么它可以作为一个边长为 4 的正方形的右下角，f 值至少为 4，即：

f[i][j]>=min(f[i][j-1],f[i-1][j],f[i-1][j-1])+1

联立两个不等式可得

f[i][j]=min(f[i][j-1],f[i-1][j],f[i-1][j-1])+1

综合考虑边界情况，递推公式如下

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/count-square-submatrices-with-all-ones/solution/tong-ji-quan-wei-1-de-zheng-fang-xing-zi-ju-zhen-2/
来源：力扣（LeetCode）

class Solution {
    public int countSquares(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        int[][] dp = new int[m][n];
        int i,j;
        int ans = 0;
        for(i = 0;i<m;i++){
            for(j=0;j<n;j++){
                if(matrix[i][j] == 0) dp[i][j] = 0;
                else if(i == 0 || j == 0) dp[i][j] = matrix[i][j];//考虑边界情况
                else{
                    dp[i][j] = Math.min(Math.min(dp[i-1][j],dp[i-1][j-1]),dp[i][j-1])+1;//正常递归求解
                }
                ans += dp[i][j];
            }
        }
        return ans;
    }
}

• 空间优化
  因为dp[i][j]和dp[i-1][j],dp[i-1][j-1],dp[i][j-1]有关，只需要保存两行数据即可

class Solution {
    public int countSquares(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        int[][] dp = new int[2][n];
        int i,j;
        int ans = 0;
        for(i = 0;i<m;i++){
            for(j=0;j<n;j++){
                if(matrix[i][j] == 0) dp[i%2][j] = 0;
                else if(i == 0 || j == 0) dp[i%2][j] = matrix[i][j];
                else{
                    dp[i%2][j] = Math.min(Math.min(dp[(i-1)%2][j],dp[(i-1)%2][j-1]),dp[i%2][j-1])+1;
                }
                ans += dp[i%2][j];
            }
        }
        return ans;
    }
}