状态:看题解后AC
代码:
dp[i][j]记录当前点为总数可以额外贡献的正方形数目,这个数目恰好等于以当前点为右下角,可以构造的最大正方形的边长。
class Solution {public:int countSquares(vector<vector<int>>& matrix) {int m = matrix.size(), n = matrix[0].size();vector<vector<int>> f(m, vector<int>(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] = min(min(f[i][j-1], f[i-1][j]), f[i-1][j-1]) + 1;}ans += f[i][j];}}return ans;}};
