状态:看题解后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;
}
};