动态规划解决两个正方形的题

两个找正方形的题

力扣221：最大正方形

在一个由 0 和 1 组成的二维矩阵内，找到只包含 1 的最大正方形，并返回其面积。

示例:

输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0

输出: 4

思路

动态规划：

1. 确定数组的意义。dp[i][j]表示以i，j为右下角的只包含1的正方形的最大边长。
2. 确立边界条件。当i=0或者j=0时，正方形的最大边长只能为0或1。
3. 确立递推条件。检查矩阵中位置的值：
   - 如果值为0，则dp[i][j]为0
   - 若值为1，则该位置的dp值取决于他左边，上边，左上的值
   递推式为：dp[i][j]=min(dp[i-1][j],dp[i-1][j-1],dp[i][j-1])+1

代码

class Solution {
    public int maximalSquare(char[][] matrix) {

        if(matrix == null) return 0;

        int len = matrix.length;
        if(len == 0) return 0;
        int wid = matrix[0].length;


        int max = Integer.MIN_VALUE;
        int[][] dp = new int[len][wid];
        for(int i=0;i<len;i++) {
            for(int j=0;j<wid;j++) {
                if(matrix[i][j] == '0') {
                    dp[i][j] = 0;
                }
                else {
                    if(i == 0 || j == 0) {
                        dp[i][j] = matrix[i][j] - '0';
                    }
                    else {
                        dp[i][j] = Math.min(Math.min(dp[i-1][j], dp[i][j-1]), dp[i-1][j-1])+1;
                    }
                    max = Math.max(dp[i][j], max);
                }
            }
        }

        return max*max;
    }
}

力扣1277：统计全为1的正方形子矩阵

给你一个 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.

思路

有了上面姐妹题的动态规划思路，不难相处这题的解法。只要把安装上题做出来的dp二维数组里面的数都加起来就可以统计出全部边长的子矩阵的数目了。代码很简答就不放了。