动态规划学习-Leetcode-1277题-统计全为1的正方形子矩形个数

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

示例 1：

输入：matrix = [ [0,1,1,1], [1,1,1,1], [0,1,1,1] ] 输出：15 解释： 边长为 1
的正方形有 10 个。 边长为 2 的正方形有 4 个。 边长为 3 的正方形有 1 个。 正方形的总数 = 10 + 4 + 1 =
15.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-square-submatrices-with-all-ones
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
**

解题思路
dp[i][j]存储自身为一个正方形，然后看周围三个dp[i-1][j] dp[i][j-1] dp[i-1][j-1]的个数，取最小的个数，因为构成一个正方形要看是否有0，最小不为0则构成一个2阶正方形。类推如果最小不为1，则可以构成3阶。

class Solution {
public:
    int countSquares(vector<vector<int>>& matrix) {
    int row=matrix.size();//行数
    int col=matrix[0].size();//列数
    int ans=0;
    vector<vector<int>>dp(row+1,vector<int>(col+1,0));
    for(int i=0;i<row;i++)
    {
        for(int j=0;j<col;j++)
        {
            if(i==0||j==0)dp[i][j]=matrix[i][j];//第一行第一列
            if(matrix[i][j]==1)
            {
                dp[i+1][j+1]=min(min(dp[i+1][j],dp[i][j+1]),dp[i][j])+1;
                //这个正方形里最小边数的dp数据+1 （自身是一个正方形）；
                ans+=dp[i+1][j+1];
            }
        }
    }
    return ans;
}
};