Leetcode 1277. Count Square Submatrices with All Ones (python + cpp)

努利！奋斗！

于 2021-01-06 09:03:46 发布

阅读量477

点赞数

分类专栏： Leetcode 动态规划文章标签： c++ 算法 leetcode python

本文链接：https://blog.csdn.net/qq_37821701/article/details/112257247

版权

本博客主要介绍LeetCode第1277题的解决方案，使用动态规划的方法。题目要求计算二维矩阵中全1子矩阵的数量。在Python和C++两种语言中实现此算法。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述

解法：动态规划

其实这道题目跟221 maximum square. 那道题里面dp数组每个位置代表以这个位置作为右下角点的最大square的边长多大。而这边只需要把那个dp数组全部加起来就是答案了。因为dp每个位置的值代表以这个位置作为右下角的点的square有多少个

python版本

class Solution:
    def countSquares(self, matrix: List[List[int]]) -> int:
        m = len(matrix)
        n = len(matrix[0])
        
        dp = [[0]*n for _ in range(m)]
        
        for i in range(m):
            if matrix[i][0] == 1:
                dp[i][0] = 1
        for i in range(n):
            if matrix[0