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

Posted 2021-08-27 数据结构和算法

截止到目前我已经写了 500多道算法题，其中部分已经整理成了pdf文档，目前总共有1000多页（并且还会不断的增加），大家可以免费下载
下载链接：https://pan.baidu.com/s/1hjwK0ZeRxYGB8lIkbKuQgQ
提取码：6666

这题和《530，动态规划解最大正方形》解法类似，不过不同的是第530题让求的是最大正方形的面积，而这题要求的是正方形的个数。我们还按照第530题的方式来解

public int countSquares(int[][] matrix) {
    int count = 0;//正方形的个数
    int[][] dp = new int[matrix.length + 1][matrix[0].length + 1];
    for (int i = 0; i < matrix.length; i++) {
        for (int j = 0; j < matrix[0].length; j++) {
            //如果当前坐标是0，就不可能构成正方形，直接跳过
            if (matrix[i][j] == 0)
                continue;
            //递推公式
            dp[i + 1][j + 1] = Math.min(Math.min(dp[i + 1][j], dp[i][j + 1]), dp[i][j]) + 1;
            //累加所有的dp值
            count += dp[i + 1][j + 1];
        }
    }
    return count;
}

总结

搞懂了《530，动态规划解最大正方形》，这题就非常简单了，这里需要理解的是以坐标(i,j)为右下角的最大正方形边长就是以(i,j)为右下角正方形的个数。