LeetCode 1277. 统计全为 1 的正方形子矩阵

原题目：https://leetcode-cn.com/problems/count-square-submatrices-with-all-ones/

 

思路：

如果边长是n，那么他可以构成n个正方形。见下图

 

代码：

class Solution {
public:
    int countSquares(vector>& matrix) {
        if(matrix.size() == 0 || matrix[0].size()==0) return 0;
        int row=matrix.size(),col = matrix[0].size();
        int sum = 0;
        vector>dp(row,vector(col));
        for(int i=0;i