LeetCode 1277 DP

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

/**
     * 给你一个 m * n 的矩阵，矩阵中的元素不是 0 就是 1，请你统计并返回其中完全由 1 组成的 正方形 子矩阵的个数。
     * @param matrix
     * @return
     */
    public static int countSquares(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        int[][] f = new int[m][n];
        int ans = 0;
        for (int i = 0; i < m; i++){
            for (int j = 0; j < n; j++) {
                if (i==0 || j==0){
                    f[i][j] = matrix[i][j];
                }else if (matrix[i][j] == 0) {
                    f[i][j] = 0;
                }else {
                    f[i][j] = Math.min(Math.min(f[i][j-1],f[i-1][j]),f[i-1][j-1])+1;
                }
                ans += f[i][j];
            }
        }
        return ans;
    }