地址:1277. 统计全为 1 的正方形子矩阵

    状态:看题解后AC

    代码:
    dp[i][j]记录当前点为总数可以额外贡献的正方形数目,这个数目恰好等于以当前点为右下角,可以构造的最大正方形的边长。

    1. class Solution {
    2. public:
    3. int countSquares(vector<vector<int>>& matrix) {
    4. int m = matrix.size(), n = matrix[0].size();
    5. vector<vector<int>> f(m, vector<int>(n));
    6. int ans = 0;
    7. for (int i = 0; i < m; ++i) {
    8. for (int j = 0; j < n; ++j) {
    9. if (i == 0 || j == 0) {
    10. f[i][j] = matrix[i][j];
    11. }
    12. else if (matrix[i][j] == 0) {
    13. f[i][j] = 0;
    14. }
    15. else {
    16. f[i][j] = min(min(f[i][j-1], f[i-1][j]), f[i-1][j-1]) + 1;
    17. }
    18. ans += f[i][j];
    19. }
    20. }
    21. return ans;
    22. }
    23. };