LeetCode 1277 DP

最新推荐文章于 2024-07-22 14:16:46 发布

原创最新推荐文章于 2024-07-22 14:16:46 发布 · 282 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于统计一个由0和1构成的矩阵中，完全由1组成的正方形子矩阵的数量。通过动态规划的方法，算法能够有效地遍历矩阵，计算并返回所有符合条件的子矩阵个数。

题目：给你一个 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;
    }