【统计全为 1 的正方形子矩阵】python刷题记录

R3-分治篇

 

class Solution:
    def countSquares(self, matrix: List[List[int]]) -> int:
        row=len(matrix)
        col=len(matrix[0])
        dp=[[0]*(col+1) for _ in range(row+1)]
        ret=0
        for i in range(row):
            for j in range(col):
                if matrix[i][j]==1:
                    dp[i+1][j+1]=min(dp[i][j+1],dp[i+1][j],dp[i][j])+1
                    ret+=dp[i+1][j+1]
        return ret