leetcode 221 最大正方形

https://leetcode-cn.com/problems/maximal-square/

题目
在一个由 0 和 1 组成的二维矩阵内，找到只包含 1 的最大正方形，并返回其面积。
示例

输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
输出: 4

思路
动态规划，与上题转移方程一样，都是 （i, j） 处代表以其为右下角延伸出的最大的符合条件的正方形
//上题转移方程写错已改正
代码

#include <cmath>
class Solution {
public:
    int maximalSquare(vector<vector<char>>& matrix) {
        int ans = 0;
        int m = matrix.size();
        int n = 0;
        if (m) n = matrix[0].size();

        vector <vector <int>> dp(m, vector<int>(n, 0));
        for (int i = 0; i < m; i++) {
            dp[i][0] = matrix[i][0] - 48;
            ans = max(ans, dp[i][0]);
        }
            
        for (int i = 0; i < n; i++) {
            dp[0][i] = matrix[0][i] - 48;
            ans = max(ans, dp[0][i]);
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                dp[i][j] = matrix[i][j] == '1' ? min(min(dp[i - 1][j - 1], dp[i][j - 1]), dp[i - 1][j]) + 1 : 0;
                ans = max(ans, dp[i][j]);
            }
        }
        return ans * ans;
    }
};